| { | |
| "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 | |
| } |