_target_: model.diffusion.corruption.multi_corruption.MultiCorruption
sdes:
  pos:
    _target_: model.common.diffusion.corruption.NumAtomsVarianceAdjustedWrappedVESDE
    wrapping_boundary: 1.0
    sigma_max: 5.0
    limit_info_key: num_atoms

  cell:
    _target_: model.common.diffusion.corruption.LatticeVPSDE.from_vpsde_config
    vpsde_config:
      beta_min: 0.1
      beta_max: 20
      limit_density: ${data_module.average_density}
      limit_var_scaling_constant: 0.25

discrete_corruptions:
  atomic_numbers:
    _target_: model.diffusion.corruption.d3pm_corruption.D3PMCorruption
    offset: 1
    d3pm:
      _target_: model.diffusion.d3pm.d3pm.MaskDiffusion
      dim: 101
      schedule:
        _target_: model.diffusion.d3pm.d3pm.create_discrete_diffusion_schedule
        kind: standard
        num_steps: 1000