Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -76,7 +76,7 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
|
|
76 |
good_vae=good_vae, # Assuming good_vae is defined elsewhere
|
77 |
joint_attention_kwargs=joint_attention_kwargs, # Fixed parameter name
|
78 |
):
|
79 |
-
yield img, seed
|
80 |
finally:
|
81 |
# Unload LoRA weights if they were loaded
|
82 |
if lora_id:
|
|
|
76 |
good_vae=good_vae, # Assuming good_vae is defined elsewhere
|
77 |
joint_attention_kwargs=joint_attention_kwargs, # Fixed parameter name
|
78 |
):
|
79 |
+
yield img, seed, seed
|
80 |
finally:
|
81 |
# Unload LoRA weights if they were loaded
|
82 |
if lora_id:
|