BackTo2014 commited on
Commit
893185d
·
verified ·
1 Parent(s): 4884249

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +8 -8
config.json CHANGED
@@ -2,16 +2,16 @@
2
  "in_channels": 3,
3
  "out_channels": 3,
4
  "down_block_types": [
5
- "ResBlock", "ResBlock", "DownSample",
6
- "ResBlock", "AttnResBlock", "ResBlock", "DownSample",
7
- "ResBlock", "ResBlock", "DownSample",
8
- "ResBlock", "ResBlock"
9
  ],
10
  "up_block_types": [
11
- "ResBlock", "ResBlock",
12
- "ResBlock", "ResBlock", "DownSample",
13
- "ResBlock", "AttnResBlock", "ResBlock", "DownSample",
14
- "ResBlock", "ResBlock", "DownSample"
15
  ],
16
  "block_out_channels": [
17
  128, 256, 256, 256
 
2
  "in_channels": 3,
3
  "out_channels": 3,
4
  "down_block_types": [
5
+ ["ResBlock", "ResBlock", "DownSample"],
6
+ ["ResBlock", "AttnResBlock", "ResBlock", "DownSample"],
7
+ ["ResBlock", "ResBlock", "DownSample"],
8
+ ["ResBlock", "ResBlock"]
9
  ],
10
  "up_block_types": [
11
+ ["ResBlock", "ResBlock"],
12
+ ["ResBlock", "ResBlock", "DownSample"],
13
+ ["ResBlock", "AttnResBlock", "ResBlock", "DownSample"],
14
+ ["ResBlock", "ResBlock", "DownSample"]
15
  ],
16
  "block_out_channels": [
17
  128, 256, 256, 256