configs/model/preconditioning/edm.yaml

_partial_: true
_target_: models.preconditioning.EDMPrecond
label_dim: ${data.label_dim} 
sigma_min: 0 
sigma_max: !!float .inf
sigma_data: 0.5