Azure Builds - OS Disk NVME Ephemeral

I am trying to get Packer to create a Generation 2 Image that has a NVME Disk for the OS. There should be a way of forcing it in Packer. Any suggestions would be appreciated.