Commit
•
7fb9e28
1
Parent(s):
4f381ea
Update app.py
Browse files
app.py
CHANGED
@@ -61,7 +61,7 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
|
|
61 |
joint_attention_kwargs={"scale": lora_scale},
|
62 |
).images[0]
|
63 |
|
64 |
-
yield image
|
65 |
|
66 |
pipe.unload_lora_weights()
|
67 |
|
@@ -118,7 +118,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
|
|
118 |
triggers=[generate_button.click, prompt.submit],
|
119 |
fn=run_lora,
|
120 |
inputs=[prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale],
|
121 |
-
outputs=[result]
|
122 |
)
|
123 |
|
124 |
app.queue()
|
|
|
61 |
joint_attention_kwargs={"scale": lora_scale},
|
62 |
).images[0]
|
63 |
|
64 |
+
yield image, seed
|
65 |
|
66 |
pipe.unload_lora_weights()
|
67 |
|
|
|
118 |
triggers=[generate_button.click, prompt.submit],
|
119 |
fn=run_lora,
|
120 |
inputs=[prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale],
|
121 |
+
outputs=[result, seed]
|
122 |
)
|
123 |
|
124 |
app.queue()
|