Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
bf42f55
1
Parent(s):
a3f28a3
Update app.py
Browse files
app.py
CHANGED
@@ -82,8 +82,8 @@ def run_lora(prompt, negative, weight, selected_state):
|
|
82 |
pipe.unet,
|
83 |
for_inference=True,
|
84 |
)
|
85 |
-
lora_model.to(device)
|
86 |
lora_model.apply_to(pipe.text_encoder, pipe.unet) #is apply too all you need?
|
|
|
87 |
#lora_model.merge_to(
|
88 |
# pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"
|
89 |
#)
|
|
|
82 |
pipe.unet,
|
83 |
for_inference=True,
|
84 |
)
|
|
|
85 |
lora_model.apply_to(pipe.text_encoder, pipe.unet) #is apply too all you need?
|
86 |
+
lora_model.to(device)
|
87 |
#lora_model.merge_to(
|
88 |
# pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"
|
89 |
#)
|