Spaces:
Paused
Paused
Commit
•
e5502cf
1
Parent(s):
722e096
Update app.py
Browse files
app.py
CHANGED
@@ -83,8 +83,7 @@ def run_lora(prompt, negative, weight, selected_state):
|
|
83 |
pipe.unet,
|
84 |
for_inference=True,
|
85 |
)
|
86 |
-
lora_model
|
87 |
-
lora_model.apply_to(pipe.text_encoder, pipe.unet) #is apply too all you need?
|
88 |
lora_model = lora_model.to("cuda").to(dtype=torch.float16)
|
89 |
|
90 |
last_merged = True
|
|
|
83 |
pipe.unet,
|
84 |
for_inference=True,
|
85 |
)
|
86 |
+
lora_model.apply_to(pipe.text_encoder, pipe.unet)
|
|
|
87 |
lora_model = lora_model.to("cuda").to(dtype=torch.float16)
|
88 |
|
89 |
last_merged = True
|