Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -57,7 +57,7 @@ def main(ref_filepath, prompt, use_nf4, progress=gr.Progress(track_tqdm=True)):
|
|
| 57 |
generator=torch.Generator("cpu").manual_seed(0),
|
| 58 |
).images[0]
|
| 59 |
filename = "output_"
|
| 60 |
-
filename += "_4bit" if
|
| 61 |
image.save(f"{filename}.png")
|
| 62 |
return f"{filename}.png", control_image
|
| 63 |
|
|
|
|
| 57 |
generator=torch.Generator("cpu").manual_seed(0),
|
| 58 |
).images[0]
|
| 59 |
filename = "output_"
|
| 60 |
+
filename += "_4bit" if use_nf4 else ""
|
| 61 |
image.save(f"{filename}.png")
|
| 62 |
return f"{filename}.png", control_image
|
| 63 |
|