diffusers
/

ddpm_dummy

Model card Files Files and versions

patrickvonplaten commited on Feb 8, 2023

Commit

a04ae6d

·

1 Parent(s): bd6c7bc

Update config.json

Files changed (1) hide show

config.json +3 -3

config.json CHANGED Viewed

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNetModel",
-  "down_blocks": ["UNetResDownBlock2D", "UNetResAttnDownBlock2D"],
-  "up_blocks": ["UNetResAttnUpBlock2D", "UNetResUpBlock2D"],
   "attn_resolutions": [
     16
   ],
@@ -9,7 +9,7 @@
   "downsample_padding": 0,
   "num_head_channels": 64,
   "ch": 32,
-  "block_channels": [32, 64],
   "ch_mult": [
     1,
     2

 {
   "_class_name": "UNetModel",
+  "down_block_types": ["DownBlock2D", "AttnDownBlock2D"],
+  "up_block_types": ["AttnUpBlock2D", "UpBlock2D"],
   "attn_resolutions": [
     16
   ],
   "downsample_padding": 0,
   "num_head_channels": 64,
   "ch": 32,
+  "block_out_channels": [32, 64],
   "ch_mult": [
     1,
     2