Krebzonide commited on
Commit
5a5ee63
·
1 Parent(s): ab6e9a4

Update app.py

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