multimodalart HF staff commited on
Commit
95108a9
1 Parent(s): caf6837

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -172,7 +172,7 @@ def run_lora(prompt, negative, lora_scale, selected_state, sdxl_loras, sdxl_lora
172
  pipe.load_lora_weights(lcm_lora_id, weight_name="lcm_sdxl_lora.safetensors", adapter_name="lcm_lora", use_auth_token=os.getenv('HF_TOKEN'))
173
  pipe.set_adapters(["loaded_lora", "lcm_lora"], adapter_weights=[0.8, 1.0])
174
  pipe.fuse_lora()
175
- pipe.disable_lora()
176
  last_fused = True
177
  is_pivotal = sdxl_loras[selected_state.index]["is_pivotal"]
178
  if(is_pivotal):
 
172
  pipe.load_lora_weights(lcm_lora_id, weight_name="lcm_sdxl_lora.safetensors", adapter_name="lcm_lora", use_auth_token=os.getenv('HF_TOKEN'))
173
  pipe.set_adapters(["loaded_lora", "lcm_lora"], adapter_weights=[0.8, 1.0])
174
  pipe.fuse_lora()
175
+ pipe.unload_lora_weights()
176
  last_fused = True
177
  is_pivotal = sdxl_loras[selected_state.index]["is_pivotal"]
178
  if(is_pivotal):