zwx00 commited on
Commit ·
98185b2
1
Parent(s): 3e89d49
try with new params to vae
Browse files- vae/config.json +2 -0
vae/config.json
CHANGED
|
@@ -22,6 +22,8 @@
|
|
| 22 |
"out_channels": 3,
|
| 23 |
"sample_size": 512,
|
| 24 |
"scaling_factor": 0.18215,
|
|
|
|
|
|
|
| 25 |
"up_block_types": [
|
| 26 |
"UpDecoderBlock2D",
|
| 27 |
"UpDecoderBlock2D",
|
|
|
|
| 22 |
"out_channels": 3,
|
| 23 |
"sample_size": 512,
|
| 24 |
"scaling_factor": 0.18215,
|
| 25 |
+
"low_cpu_mem_usage": false,
|
| 26 |
+
"device_map": null,
|
| 27 |
"up_block_types": [
|
| 28 |
"UpDecoderBlock2D",
|
| 29 |
"UpDecoderBlock2D",
|