Spaces:
Runtime error
Runtime error
Commit
·
44f2b12
1
Parent(s):
f7d0728
Update app.py
Browse files
app.py
CHANGED
|
@@ -74,7 +74,7 @@ def run_lora(prompt, negative, weight, selected_state):
|
|
| 74 |
else:
|
| 75 |
multiplier = 1.0
|
| 76 |
pipe.to("cpu")
|
| 77 |
-
|
| 78 |
lora_model, weights_sd = lora.create_network_from_weights(
|
| 79 |
multiplier,
|
| 80 |
full_path_lora,
|
|
|
|
| 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,
|
| 80 |
full_path_lora,
|