BackTo2014 commited on
Commit
5bf34e0
1 Parent(s): 63c3882

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
- "DownSample",
6
- "DownSample",
7
- "DownSample",
8
- "ResBlock"
9
  ],
10
  "up_block_types": [
11
- "ResBlock",
12
- "DownSample",
13
- "DownSample",
14
- "DownSample"
15
  ],
16
  "block_out_channels": [
17
  128, 256, 256, 256
 
2
  "in_channels": 3,
3
  "out_channels": 3,
4
  "down_block_types": [
5
+ "DownBlock2D",
6
+ "DownBlock2D",
7
+ "AttnDownBlock2D",
8
+ "DownBlock2D"
9
  ],
10
  "up_block_types": [
11
+ "DownBlock2D",
12
+ "AttnUpBlock2D",
13
+ "DownBlock2D",
14
+ "DownBlock2D"
15
  ],
16
  "block_out_channels": [
17
  128, 256, 256, 256