Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -162,9 +162,15 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength, cfg_scale, s
|
|
| 162 |
seed = random.randint(0, MAX_SEED)
|
| 163 |
|
| 164 |
# Load LoRA weights
|
| 165 |
-
gr.Info("Start to load
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 166 |
lora_configs = None
|
| 167 |
adapter_names = []
|
|
|
|
| 168 |
if lora_strings_json:
|
| 169 |
try:
|
| 170 |
lora_configs = json.loads(lora_strings_json)
|
|
|
|
| 162 |
seed = random.randint(0, MAX_SEED)
|
| 163 |
|
| 164 |
# Load LoRA weights
|
| 165 |
+
gr.Info("Start to load LoRA ...")
|
| 166 |
+
|
| 167 |
+
with calculateDuration("Unloading LoRA"):
|
| 168 |
+
pipe.unload_lora_weights()
|
| 169 |
+
img2img.unload_lora_weights()
|
| 170 |
+
|
| 171 |
lora_configs = None
|
| 172 |
adapter_names = []
|
| 173 |
+
|
| 174 |
if lora_strings_json:
|
| 175 |
try:
|
| 176 |
lora_configs = json.loads(lora_strings_json)
|