multimodalart HF staff commited on
Commit
43f359d
β€’
1 Parent(s): a8d5a97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -149,14 +149,15 @@ 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
- 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.unload_lora_weights()
159
- pipe.unfuse_lora()
 
160
  is_compatible = sdxl_loras[selected_state.index]["is_compatible"]
161
  if is_compatible:
162
  pipe.load_lora_weights(loaded_state_dict)
 
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
+
158
+ #elif(last_fused):
159
+ # pipe.unload_lora_weights()
160
+ # pipe.unfuse_lora()
161
  is_compatible = sdxl_loras[selected_state.index]["is_compatible"]
162
  if is_compatible:
163
  pipe.load_lora_weights(loaded_state_dict)