Spaces:
Running
on
A100
Running
on
A100
Commit
•
6f329ae
1
Parent(s):
9bbaa16
Update app.py
Browse files
app.py
CHANGED
@@ -118,6 +118,9 @@ def merge_incompatible_lora(full_path_lora, lora_scale):
|
|
118 |
lora_model.merge_to(
|
119 |
pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"
|
120 |
)
|
|
|
|
|
|
|
121 |
|
122 |
def run_lora(prompt, negative, lora_scale, selected_state):
|
123 |
global last_lora, last_merged, pipe
|
|
|
118 |
lora_model.merge_to(
|
119 |
pipe.text_encoder, pipe.unet, weights_sd, torch.float16, "cuda"
|
120 |
)
|
121 |
+
del weights_sd
|
122 |
+
del lora_model
|
123 |
+
gc.collect()
|
124 |
|
125 |
def run_lora(prompt, negative, lora_scale, selected_state):
|
126 |
global last_lora, last_merged, pipe
|