Spaces:
Build error
Build error
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()
|