noise_scheduler_kwargs:
  beta_start:    0.0001
  beta_end:      0.02
  beta_schedule: "linear"
  steps_offset:  1

vae_kwargs:
  enable_magvit: true
  slice_compression_vae: true