Spaces:
Running
on
A100
Running
on
A100
Commit
•
06f770c
1
Parent(s):
f424501
Update app.py
Browse files
app.py
CHANGED
@@ -158,6 +158,7 @@ def run_lora(prompt, negative, lora_scale, selected_state):
|
|
158 |
cross_attention_kwargs=cross_attention_kwargs,
|
159 |
).images[0]
|
160 |
last_lora = repo_name
|
|
|
161 |
return image, gr.update(visible=True)
|
162 |
|
163 |
|
|
|
158 |
cross_attention_kwargs=cross_attention_kwargs,
|
159 |
).images[0]
|
160 |
last_lora = repo_name
|
161 |
+
gc.collect()
|
162 |
return image, gr.update(visible=True)
|
163 |
|
164 |
|