_target_: src.envs.WorldModelEnvConfig
horizon: 1000
num_batches_to_preload: 1
diffusion_sampler_next_obs:
  _target_: src.models.diffusion.DiffusionSamplerConfig
  num_steps_denoising: 6  # Balanced: better quality than 3, faster than 10
  sigma_min: 0.002
  sigma_max: 5.0
  rho: 7
  order: 1
diffusion_sampler_upsampling:
  _target_: src.models.diffusion.DiffusionSamplerConfig
  num_steps_denoising: 4  # Balanced: better quality than 2, faster than 5
  sigma_min: 0.002
  sigma_max: 5.0
  rho: 7
  order: 1