openai
/

diffusers-cd_imagenet64_l2

ConsistencyModelPipeline

generative model

unconditional image generation

consistency-model

Model card Files Files and versions Community

patrickvonplaten commited on Jul 5, 2023

Commit

c44bb87

•

1 Parent(s): c3795c9

Update unet/config.json

Files changed (1) hide show

unet/config.json +8 -6

unet/config.json CHANGED Viewed

@@ -15,9 +15,9 @@
   "class_embed_type": null,
   "down_block_types": [
     "ResnetDownsampleBlock2D",
-    "AttnDownsampleBlock2D",
-    "AttnDownsampleBlock2D",
-    "AttnDownsampleBlock2D"
   ],
   "downsample_padding": 1,
   "flip_sin_to_cos": true,
@@ -32,10 +32,12 @@
   "resnet_time_scale_shift": "scale_shift",
   "sample_size": 64,
   "time_embedding_type": "positional",
   "up_block_types": [
-    "AttnUpsampleBlock2D",
-    "AttnUpsampleBlock2D",
-    "AttnUpsampleBlock2D",
     "ResnetUpsampleBlock2D"
   ]
 }

   "class_embed_type": null,
   "down_block_types": [
     "ResnetDownsampleBlock2D",
+    "AttnDownBlock2D",
+    "AttnDownBlock2D",
+    "AttnDownBlock2D"
   ],
   "downsample_padding": 1,
   "flip_sin_to_cos": true,
   "resnet_time_scale_shift": "scale_shift",
   "sample_size": 64,
   "time_embedding_type": "positional",
+  "upsample_type": "resnet",
+  "downsample_type": "resnet",
   "up_block_types": [
+    "AttnUpBlock2D",
+    "AttnUpBlock2D",
+    "AttnUpBlock2D",
     "ResnetUpsampleBlock2D"
   ]
 }