multimodalart HF staff commited on
Commit
d77a82f
β€’
1 Parent(s): e862e78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -149,14 +149,14 @@ def run_lora(prompt, negative, lora_scale, selected_state, progress=gr.Progress(
149
  loaded_state_dict = sdxl_loras[selected_state.index]["state_dict"]
150
  cross_attention_kwargs = None
151
  if last_lora != repo_name:
152
- pipe = copy.deepcopy(original_pipe)
153
- pipe.to(device)
154
  if last_merged:
155
  del pipe
156
  gc.collect()
157
- #elif(last_fused):
158
- # pipe.unfuse_lora()
159
- # pipe.unload_lora_weights()
 
 
160
  is_compatible = sdxl_loras[selected_state.index]["is_compatible"]
161
 
162
  if is_compatible:
 
149
  loaded_state_dict = sdxl_loras[selected_state.index]["state_dict"]
150
  cross_attention_kwargs = None
151
  if last_lora != repo_name:
 
 
152
  if last_merged:
153
  del pipe
154
  gc.collect()
155
+ pipe = copy.deepcopy(original_pipe)
156
+ pipe.to(device)
157
+ elif(last_fused):
158
+ pipe.unfuse_lora()
159
+ pipe.unload_lora_weights()
160
  is_compatible = sdxl_loras[selected_state.index]["is_compatible"]
161
 
162
  if is_compatible: