Update config.yaml
Browse files- config.yaml +2 -4
config.yaml
CHANGED
|
@@ -33,9 +33,7 @@ model:
|
|
| 33 |
in_channels: 4
|
| 34 |
out_channels: 4
|
| 35 |
model_channels: 4
|
| 36 |
-
|
| 37 |
-
# num_res_blocks: 2
|
| 38 |
-
channel_mult: [ 1, 2]
|
| 39 |
num_heads: 8
|
| 40 |
use_spatial_transformer: True
|
| 41 |
transformer_depth: 1
|
|
@@ -58,7 +56,7 @@ model:
|
|
| 58 |
ch_mult:
|
| 59 |
- 1
|
| 60 |
- 2
|
| 61 |
-
|
| 62 |
attn_resolutions: []
|
| 63 |
dropout: 0.0
|
| 64 |
lossconfig:
|
|
|
|
| 33 |
in_channels: 4
|
| 34 |
out_channels: 4
|
| 35 |
model_channels: 4
|
| 36 |
+
channel_mult: [ 1, 2 ]
|
|
|
|
|
|
|
| 37 |
num_heads: 8
|
| 38 |
use_spatial_transformer: True
|
| 39 |
transformer_depth: 1
|
|
|
|
| 56 |
ch_mult:
|
| 57 |
- 1
|
| 58 |
- 2
|
| 59 |
+
num_res_blocks: 2
|
| 60 |
attn_resolutions: []
|
| 61 |
dropout: 0.0
|
| 62 |
lossconfig:
|