config.json

{
  "architecture": "ConditionalUNet + GaussianDiffusion",
  "task": "conditional-image-generation",
  "image_size": 256,
  "in_channels": 1,
  "out_channels": 1,
  "label_dim": 2,
  "base_channels": 64,
  "channel_multipliers": [
    1,
    2,
    4,
    8
  ],
  "attention_levels": [
    2,
    3
  ],
  "dropout": 0.1,
  "timesteps": 1500,
  "beta_start": 0.0001,
  "beta_end": 0.02,
  "schedule_type": "linear"
}