Commit
·
5b12685
1
Parent(s):
97b4689
Update app.py
Browse files
app.py
CHANGED
@@ -86,9 +86,9 @@ def run_lora(prompt, negative, weight, selected_state):
|
|
86 |
lora_model.to(device)
|
87 |
lora_model.apply_to(pipe.text_encoder, pipe.unet) #is apply too all you need?
|
88 |
lora_model.to(device)
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
pipe.to(device)
|
93 |
last_merged = True
|
94 |
|
|
|
86 |
lora_model.to(device)
|
87 |
lora_model.apply_to(pipe.text_encoder, pipe.unet) #is apply too all you need?
|
88 |
lora_model.to(device)
|
89 |
+
lora_model.merge_to(
|
90 |
+
pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"
|
91 |
+
)
|
92 |
pipe.to(device)
|
93 |
last_merged = True
|
94 |
|