{
  "model_type": "diffusion",
  "base_channels": 64,
  "in_channels": 3,
  "out_channels": 3,
  "num_classes": 10,
  "text_embed_dim": 512,
  "use_text_conditioning": true,
  "timesteps": 1000,
  "beta_start": 0.0001,
  "beta_end": 0.02,
  "ddim_steps": 50,
  "guidance_scale": 3.0
}