fffiloni commited on
Commit
7d2fd42
1 Parent(s): 08deb2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 four_bit else ""
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