Update config.yaml
Browse files- config.yaml +2 -0
config.yaml
CHANGED
|
@@ -37,6 +37,7 @@ model:
|
|
| 37 |
attention_resolutions: [2, 1]
|
| 38 |
num_heads: 8
|
| 39 |
num_res_blocks: 1
|
|
|
|
| 40 |
use_spatial_transformer: True
|
| 41 |
transformer_depth: 1
|
| 42 |
context_dim: 32
|
|
@@ -59,6 +60,7 @@ model:
|
|
| 59 |
- 1
|
| 60 |
- 2
|
| 61 |
num_res_blocks: 1
|
|
|
|
| 62 |
attn_resolutions: []
|
| 63 |
dropout: 0.0
|
| 64 |
lossconfig:
|
|
|
|
| 37 |
attention_resolutions: [2, 1]
|
| 38 |
num_heads: 8
|
| 39 |
num_res_blocks: 1
|
| 40 |
+
norm_num_groups: 2
|
| 41 |
use_spatial_transformer: True
|
| 42 |
transformer_depth: 1
|
| 43 |
context_dim: 32
|
|
|
|
| 60 |
- 1
|
| 61 |
- 2
|
| 62 |
num_res_blocks: 1
|
| 63 |
+
norm_num_groups: 2
|
| 64 |
attn_resolutions: []
|
| 65 |
dropout: 0.0
|
| 66 |
lossconfig:
|