Update unet/config.json
Browse files- unet/config.json +1 -1
unet/config.json
CHANGED
|
@@ -5,7 +5,7 @@
|
|
| 5 |
"act_fn": "silu",
|
| 6 |
"attention_head_dim": 8,
|
| 7 |
"block_out_channels": [
|
| 8 |
-
|
| 9 |
512,
|
| 10 |
512,
|
| 11 |
1024
|
|
|
|
| 5 |
"act_fn": "silu",
|
| 6 |
"attention_head_dim": 8,
|
| 7 |
"block_out_channels": [
|
| 8 |
+
320,
|
| 9 |
512,
|
| 10 |
512,
|
| 11 |
1024
|