Krebzonide
commited on
Commit
·
5a5ee63
1
Parent(s):
ab6e9a4
Update app.py
Browse files
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+
|
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)
|