jiuface commited on
Commit
3fefa6e
·
verified ·
1 Parent(s): cc9f7a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -166,9 +166,7 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength, cfg_scale, s
166
  gr.Info("Start to load LoRA ...")
167
  with calculateDuration("Unloading LoRA"):
168
  img2img_pipe.unload_lora_weights()
169
- # img2img_pipe.unload_lora()
170
  txt2img_pipe.unload_lora_weights()
171
- txt2img_pipe.unload_lora()
172
 
173
  print(txt2img_pipe.get_active_adapters())
174
  list_adapters_component_wise = txt2img_pipe.get_list_adapters()
@@ -237,7 +235,9 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength, cfg_scale, s
237
  result = {"status": "success", "message": "Image generated but not uploaded"}
238
  else:
239
  result = {"status": "failed", "message": error_message}
240
-
 
 
241
  gr.Info("Completed!")
242
  progress(100, "Completed!")
243
  torch.cuda.empty_cache()
 
166
  gr.Info("Start to load LoRA ...")
167
  with calculateDuration("Unloading LoRA"):
168
  img2img_pipe.unload_lora_weights()
 
169
  txt2img_pipe.unload_lora_weights()
 
170
 
171
  print(txt2img_pipe.get_active_adapters())
172
  list_adapters_component_wise = txt2img_pipe.get_list_adapters()
 
235
  result = {"status": "success", "message": "Image generated but not uploaded"}
236
  else:
237
  result = {"status": "failed", "message": error_message}
238
+
239
+ txt2img_pipe.delete_adapters(lora_names)
240
+
241
  gr.Info("Completed!")
242
  progress(100, "Completed!")
243
  torch.cuda.empty_cache()