Update unet/config.json
Browse files- unet/config.json +1 -1
unet/config.json
CHANGED
|
@@ -32,7 +32,7 @@
|
|
| 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",
|
|
|
|
| 32 |
"mid_block_scale_factor": 1,
|
| 33 |
"norm_eps": 1e-05,
|
| 34 |
"norm_num_groups": 32,
|
| 35 |
+
"out_channels": 8,
|
| 36 |
"sample_size": 96,
|
| 37 |
"up_block_types": [
|
| 38 |
"UpBlock2D",
|