multimodalart HF staff commited on
Commit
a675513
1 Parent(s): b4300b3

Update app.py

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