primerz commited on
Commit
99d58c2
·
verified ·
1 Parent(s): 0775e46

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +2 -1
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
- pipe.load_lora_weights(lora_path)
 
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: