Spaces:
Runtime error
Runtime error
Update models.py
Browse files
models.py
CHANGED
|
@@ -164,7 +164,8 @@ def load_lora(pipe):
|
|
| 164 |
print("Loading LORA (retroart) from HuggingFace Hub...")
|
| 165 |
try:
|
| 166 |
lora_path = download_model_with_retry(MODEL_REPO, MODEL_FILES['lora'])
|
| 167 |
-
|
|
|
|
| 168 |
print(f" [OK] LORA loaded successfully")
|
| 169 |
return True
|
| 170 |
except Exception as e:
|
|
|
|
| 164 |
print("Loading LORA (retroart) from HuggingFace Hub...")
|
| 165 |
try:
|
| 166 |
lora_path = download_model_with_retry(MODEL_REPO, MODEL_FILES['lora'])
|
| 167 |
+
# **FIX 2: Add adapter_name="retroart"**
|
| 168 |
+
pipe.load_lora_weights(lora_path, adapter_name="retroart")
|
| 169 |
print(f" [OK] LORA loaded successfully")
|
| 170 |
return True
|
| 171 |
except Exception as e:
|