{
  "_class_name": "GaussianDiffusion",
  "beta_schedule": "cosine",
  "channels": 3,
  "image_size": 128,
  "loss_type": "l1",
  "objective": "pred_noise",
  "timesteps": 3
}