ovi054 commited on
Commit
388486e
·
verified ·
1 Parent(s): c4cc217

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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: