Azure - Creating new / Gen2

What azure module can i use to set the vm generation when creating a new vm? Figured it would be under azure_rm_virtualmachine.

Would like to make sure it will be created with gen 2.