patrickvonplaten commited on
Commit
d8e456d
1 Parent(s): 18b9dba

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -4,7 +4,7 @@
4
  16
5
  ],
6
  "down_blocks": ["UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResDownBlock2D"],
7
- "up_blocks": ["UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D"],
8
  "conv_resample": true,
9
  "downsample_padding": 0,
10
  "num_head_channels": 64,
 
4
  16
5
  ],
6
  "down_blocks": ["UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResDownBlock2D"],
7
+ "up_blocks": ["UNetResUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D"],
8
  "conv_resample": true,
9
  "downsample_padding": 0,
10
  "num_head_channels": 64,