Upload config.json with huggingface_hub

config.json

@@ -0,0 +1,34 @@
+{
+  "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
+      ]
+    }
+  }
+}