multimodalart HF staff commited on
Commit
06f770c
1 Parent(s): f424501

Update app.py

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