multimodalart HF staff commited on
Commit
bf6d614
1 Parent(s): 44f2b12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ def run_lora(prompt, negative, weight, selected_state):
73
  multiplier = float(weight)
74
  else:
75
  multiplier = 1.0
76
- pipe.to("cpu")
77
  multiplier = torch.tensor([multiplier], dtype=torch.float16, device=device)
78
  lora_model, weights_sd = lora.create_network_from_weights(
79
  multiplier,
 
73
  multiplier = float(weight)
74
  else:
75
  multiplier = 1.0
76
+
77
  multiplier = torch.tensor([multiplier], dtype=torch.float16, device=device)
78
  lora_model, weights_sd = lora.create_network_from_weights(
79
  multiplier,