patrickvonplaten commited on
Commit
5d64f40
·
1 Parent(s): 1653231

Update unet/config.json

Browse files
Files changed (1) hide show
  1. unet/config.json +2 -2
unet/config.json CHANGED
@@ -8,8 +8,8 @@
8
  ],
9
  "down_blocks": ["UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResAttnDownBlock2D", "UNetResAttnDownBlock2D"],
10
  "up_blocks": ["UNetResAttnUpBlock2D", "UNetResAttnUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D"],
11
- "block_input_channels": [224, 224, 448, 672],
12
- "block_output_channels": [224, 448, 672, 896],
13
  "context_dim": null,
14
  "conv_resample": true,
15
  "dims": 2,
 
8
  ],
9
  "down_blocks": ["UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResAttnDownBlock2D", "UNetResAttnDownBlock2D"],
10
  "up_blocks": ["UNetResAttnUpBlock2D", "UNetResAttnUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D"],
11
+ "down_block_input_channels": [224, 224, 448, 672],
12
+ "down_block_output_channels": [224, 448, 672, 896],
13
  "context_dim": null,
14
  "conv_resample": true,
15
  "dims": 2,