Update unet/config.json

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",
@@ -45,7 +45,7 @@
   "num_class_embeds_shape": 5,
   "num_class_embeds_wall_hue": 11,
   "num_train_timesteps": null,
-  "optimization_step": 90000,
+  "optimization_step": 170000,
   "out_channels": 3,
   "power": 0.75,
   "resnet_time_scale_shift": "default",
@@ -53,15 +53,15 @@
   "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",
   "use_ema_warmup": true
-}
+}