patrickvonplaten commited on
Commit
33fd470
·
1 Parent(s): a0ffc06

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -3
config.json CHANGED
@@ -45,7 +45,6 @@
45
  "skip_rescale": true,
46
  "block_channels": [128, 128, 256, 256, 256, 256, 256],
47
  "in_channels": 3,
48
- "num_res_blocks": 1,
49
  "out_channels": 3,
50
  "time_embedding_type": "fourier",
51
  "resnet_eps": 1e-6,
@@ -56,14 +55,14 @@
56
  "UNetResSkipDownBlock2D",
57
  "UNetResSkipDownBlock2D",
58
  "UNetResSkipDownBlock2D",
59
- "UNetResSkipDownBlock2D",
60
  "UNetResAttnSkipDownBlock2D",
 
61
  "UNetResSkipDownBlock2D"
62
  ],
63
  "up_blocks": [
64
  "UNetResSkipUpBlock2D",
65
- "UNetResAttnSkipUpBlock2D",
66
  "UNetResSkipUpBlock2D",
 
67
  "UNetResSkipUpBlock2D",
68
  "UNetResSkipUpBlock2D",
69
  "UNetResSkipUpBlock2D",
 
45
  "skip_rescale": true,
46
  "block_channels": [128, 128, 256, 256, 256, 256, 256],
47
  "in_channels": 3,
 
48
  "out_channels": 3,
49
  "time_embedding_type": "fourier",
50
  "resnet_eps": 1e-6,
 
55
  "UNetResSkipDownBlock2D",
56
  "UNetResSkipDownBlock2D",
57
  "UNetResSkipDownBlock2D",
 
58
  "UNetResAttnSkipDownBlock2D",
59
+ "UNetResSkipDownBlock2D",
60
  "UNetResSkipDownBlock2D"
61
  ],
62
  "up_blocks": [
63
  "UNetResSkipUpBlock2D",
 
64
  "UNetResSkipUpBlock2D",
65
+ "UNetResAttnSkipUpBlock2D",
66
  "UNetResSkipUpBlock2D",
67
  "UNetResSkipUpBlock2D",
68
  "UNetResSkipUpBlock2D",