Spaces:
Running
on
Zero
Running
on
Zero
Restore --logdir argument now that sample.py supports it
Browse files
app.py
CHANGED
|
@@ -166,7 +166,8 @@ def generate_image(prompt, aspect_ratio, cfg, seed, checkpoint_file, config_path
|
|
| 166 |
"--finetuned_vae", "none",
|
| 167 |
"--jacobi", "1",
|
| 168 |
"--jacobi_th", "0.001",
|
| 169 |
-
"--jacobi_block_size", "16"
|
|
|
|
| 170 |
]
|
| 171 |
|
| 172 |
status_msg += "Running generation...\n"
|
|
|
|
| 166 |
"--finetuned_vae", "none",
|
| 167 |
"--jacobi", "1",
|
| 168 |
"--jacobi_th", "0.001",
|
| 169 |
+
"--jacobi_block_size", "16",
|
| 170 |
+
"--logdir", str(output_dir) # Set logdir to outputs directory
|
| 171 |
]
|
| 172 |
|
| 173 |
status_msg += "Running generation...\n"
|