{
  "model_type": "conditional_unet",
  "architecture": "ConditionalUNet",
  "in_channels": 3,
  "out_channels": 3,
  "features": 64,
  "num_colors": 8,
  "color_embed_dim": 32,
  "img_size": 128,
  "colors": [
    "blue",
    "cyan",
    "green",
    "magenta",
    "orange",
    "purple",
    "red",
    "yellow"
  ],
  "preprocessing": {
    "normalize": {
      "mean": [
        0.485,
        0.456,
        0.406
      ],
      "std": [
        0.229,
        0.224,
        0.225
      ]
    }
  }
}