Spaces:
Paused
Paused
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 |
|