Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
d21a36f
1
Parent(s):
e5502cf
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 |
|
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,
|
@@ -84,7 +84,7 @@ def run_lora(prompt, negative, weight, selected_state):
|
|
84 |
for_inference=True,
|
85 |
)
|
86 |
lora_model.apply_to(pipe.text_encoder, pipe.unet)
|
87 |
-
lora_model = lora_model.to("cuda")
|
88 |
|
89 |
last_merged = True
|
90 |
|
|
|
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,
|
|
|
84 |
for_inference=True,
|
85 |
)
|
86 |
lora_model.apply_to(pipe.text_encoder, pipe.unet)
|
87 |
+
lora_model = lora_model.to("cuda")
|
88 |
|
89 |
last_merged = True
|
90 |
|