multimodalart HF staff commited on
Commit
cd34443
1 Parent(s): bf42f55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -73,7 +73,8 @@ def run_lora(prompt, negative, weight, selected_state):
73
  multiplier = float(weight)
74
  else:
75
  multiplier = 1.0
76
-
 
77
  lora_model, weights_sd = lora.create_network_from_weights(
78
  multiplier,
79
  full_path_lora,
 
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,
80
  full_path_lora,