Commit
•
a675513
1
Parent(s):
b4300b3
Update app.py
Browse files
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,
|