multimodalart HF staff commited on
Commit
46b8e8d
β€’
1 Parent(s): a675513

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -59,6 +59,7 @@ def run_lora(prompt, negative, weight, selected_state):
59
  if last_lora != repo_name:
60
  if last_merged:
61
  pipe = copy.deepcopy(original_pipe)
 
62
  else:
63
  pipe.unload_lora_weights()
64
  is_compatible = sdxl_loras[selected_state.index][5]
 
59
  if last_lora != repo_name:
60
  if last_merged:
61
  pipe = copy.deepcopy(original_pipe)
62
+ pipe.to(device)
63
  else:
64
  pipe.unload_lora_weights()
65
  is_compatible = sdxl_loras[selected_state.index][5]