Krebzonide commited on
Commit
6518260
·
1 Parent(s): b550dc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -58,7 +58,7 @@ def generate(prompt, neg_prompt, samp_steps, cfg_scale, batch_size, seed, height
58
  width=width,
59
  generator=torch.manual_seed(seed),
60
  ).images
61
- return [(img, f"Image {i+1}") for i, img in enumerate(images), seed]
62
 
63
  def set_base_model(base_model_id):
64
  vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 
58
  width=width,
59
  generator=torch.manual_seed(seed),
60
  ).images
61
+ return [(img, f"Image {i+1}") for i, img in enumerate(images)], seed
62
 
63
  def set_base_model(base_model_id):
64
  vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)