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