Update unet/config.json
Browse files- unet/config.json +2 -2
unet/config.json
CHANGED
|
@@ -27,12 +27,12 @@
|
|
| 27 |
"dual_cross_attention": false,
|
| 28 |
"flip_sin_to_cos": true,
|
| 29 |
"freq_shift": 0,
|
| 30 |
-
"in_channels":
|
| 31 |
"layers_per_block": 2,
|
| 32 |
"mid_block_scale_factor": 1,
|
| 33 |
"norm_eps": 1e-05,
|
| 34 |
"norm_num_groups": 32,
|
| 35 |
-
"out_channels":
|
| 36 |
"sample_size": 96,
|
| 37 |
"up_block_types": [
|
| 38 |
"UpBlock2D",
|
|
|
|
| 27 |
"dual_cross_attention": false,
|
| 28 |
"flip_sin_to_cos": true,
|
| 29 |
"freq_shift": 0,
|
| 30 |
+
"in_channels": 8,
|
| 31 |
"layers_per_block": 2,
|
| 32 |
"mid_block_scale_factor": 1,
|
| 33 |
"norm_eps": 1e-05,
|
| 34 |
"norm_num_groups": 32,
|
| 35 |
+
"out_channels": 4,
|
| 36 |
"sample_size": 96,
|
| 37 |
"up_block_types": [
|
| 38 |
"UpBlock2D",
|