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