@Juan Pablo Delgadillo Martinez
I would recommend checking out this feature as it can help achieve your goal:
Announcing the public preview of Azure Virtual Desktop Custom Image Templates (microsoft.com)
Please be aware that it is currently in Public Preview:
Custom image templates allow admins to build a custom “golden image” with the added capability to include Azure Virtual Desktop built-in customizations as well as your own customization scripts to install other applications or set of configurations.
You can also use the following script to modify the Host Pool's default DiskSizeinGB value for a specific Host Pool (HP), so next VMs deployed to on that Host Pool will use the new value for disk size:
#Select desired Host Pool to update
$resource = Get-AzResource -ResourceType 'Microsoft.DesktopVirtualization/hostpools' | Out-GridView -OutputMode Single -Title 'Select the HP to change'
$myHP = Get-AzWvdHostPool -Name $($resource.Name) -ResourceGroupName $($resource.ResourceGroupName)
#import vm settings as PS Object.
$newvmtemplate = $myHP.VMTemplate | convertfrom-json
#display host pool's VM config
$newvmtemplate
#modify settings using the correct sizings e.g.
$newvmtemplate.diskSizeGB = 256
#update HP's template with new disk size
Update-AzWvdHostPool -Name $($myHP.Name) -VMTemplate $($newvmtemplate | ConvertTo-Json) -ResourceGroupName $($resource.ResourceGroupName)
Let me know if you have any questions.