Commit
·
3af8c1d
1
Parent(s):
40573e7
Update app.py
Browse files
app.py
CHANGED
|
@@ -85,6 +85,7 @@ def run_lora(prompt, negative, weight, selected_state):
|
|
| 85 |
)
|
| 86 |
lora_model.to("cuda")
|
| 87 |
lora_model.apply_to(pipe.text_encoder, pipe.unet) #is apply too all you need?
|
|
|
|
| 88 |
last_merged = True
|
| 89 |
|
| 90 |
image = pipe(
|
|
|
|
| 85 |
)
|
| 86 |
lora_model.to("cuda")
|
| 87 |
lora_model.apply_to(pipe.text_encoder, pipe.unet) #is apply too all you need?
|
| 88 |
+
lora_model.to("cuda")
|
| 89 |
last_merged = True
|
| 90 |
|
| 91 |
image = pipe(
|