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