Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
0df5d5a
1
Parent(s):
b888bcf
Update app.py
Browse files
app.py
CHANGED
|
@@ -81,10 +81,10 @@ def run_lora(prompt, negative, weight, selected_state):
|
|
| 81 |
pipe.unet,
|
| 82 |
for_inference=True,
|
| 83 |
)
|
| 84 |
-
lora_model.apply_to(pipe.text_encoder, pipe.unet)
|
| 85 |
-
lora_model.merge_to(
|
| 86 |
-
|
| 87 |
-
)
|
| 88 |
last_merged = True
|
| 89 |
|
| 90 |
image = pipe(
|
|
|
|
| 81 |
pipe.unet,
|
| 82 |
for_inference=True,
|
| 83 |
)
|
| 84 |
+
lora_model.apply_to(pipe.text_encoder, pipe.unet) #is apply too all you need?
|
| 85 |
+
#lora_model.merge_to(
|
| 86 |
+
# pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"
|
| 87 |
+
#)
|
| 88 |
last_merged = True
|
| 89 |
|
| 90 |
image = pipe(
|