Upload unet/config.json with huggingface_hub

unet/config.json

@@ -18,13 +18,13 @@
   "class_embed_type": null,
   "decay": 0.9999,
   "down_block_types": [
-    "DownBlock2D",
-    "DownBlock2D",
-    "DownBlock2D",
-    "DownBlock2D",
-    "AttnDownBlock2D",
-    "DownBlock2D",
-    "DownBlock2D"
+    "SkipDownBlock2D",
+    "SkipDownBlock2D",
+    "SkipDownBlock2D",
+    "SkipDownBlock2D",
+    "AttnSkipDownBlock2D",
+    "SkipDownBlock2D",
+    "SkipDownBlock2D"
   ],
   "downsample_padding": 1,
   "downsample_type": "conv",
@@ -37,10 +37,10 @@
   "mid_block_scale_factor": 1.41421356237,
   "min_decay": 0.0,
   "norm_eps": 1e-06,
-  "norm_num_groups": 32,
+  "norm_num_groups": null,
   "num_class_embeds": null,
   "num_train_timesteps": null,
-  "optimization_step": 130000,
+  "optimization_step": 120000,
   "out_channels": 3,
   "power": 0.6666666666666666,
   "resnet_time_scale_shift": "default",
@@ -48,13 +48,13 @@
   "set_W_to_weight": false,
   "time_embedding_type": "fourier",
   "up_block_types": [
-    "UpBlock2D",
-    "UpBlock2D",
-    "AttnUpBlock2D",
-    "UpBlock2D",
-    "UpBlock2D",
-    "UpBlock2D",
-    "UpBlock2D"
+    "SkipUpBlock2D",
+    "SkipUpBlock2D",
+    "AttnSkipUpBlock2D",
+    "SkipUpBlock2D",
+    "SkipUpBlock2D",
+    "SkipUpBlock2D",
+    "SkipUpBlock2D"
   ],
   "update_after_step": 0,
   "upsample_type": "conv",