Spaces:
Running
on
A100
Running
on
A100
Commit
•
fc8ab35
1
Parent(s):
b52bfac
Update app.py
Browse files
app.py
CHANGED
@@ -85,7 +85,7 @@ def run_lora(prompt, negative, weight, selected_state):
|
|
85 |
)
|
86 |
lora_model = lora_model.to("cuda")
|
87 |
lora_model.apply_to(pipe.text_encoder, pipe.unet)
|
88 |
-
|
89 |
last_merged = True
|
90 |
|
91 |
image = pipe(
|
|
|
85 |
)
|
86 |
lora_model = lora_model.to("cuda")
|
87 |
lora_model.apply_to(pipe.text_encoder, pipe.unet)
|
88 |
+
lora_model = lora_model.to("cuda")
|
89 |
last_merged = True
|
90 |
|
91 |
image = pipe(
|