fusing
/

ddpm-cifar10

Inference Endpoints

Model card Files Files and versions Community

patrickvonplaten commited on Jul 14, 2022

Commit

8682899

•

1 Parent(s): 5987f53

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -3,8 +3,8 @@
   "attn_resolutions": [
     16
   ],
-  "down_blocks": ["UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResDownBlock2D"],
-  "up_blocks": ["UNetResUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D"],
   "conv_resample": true,
   "downsample_padding": 0,
   "num_head_channels": 64,

   "attn_resolutions": [
     16
   ],
+  "down_blocks": ["UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D"],
+  "up_blocks": ["UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D"],
   "conv_resample": true,
   "downsample_padding": 0,
   "num_head_channels": 64,