Update lora_saver.py

lora_saver.py

@@ -9,19 +9,19 @@ HF_REPO = "K00B404/LoraStack"
 
 def download_lora(repo_id, save_dir="loras"):
     """Download all files from a LoRA repository."""
-    print(f"lora repo {repo_id.get('repo')}")
-    model_name = repo_id.get('repo').split("/")[-1]
+    print(f"lora repo {repo_id}")
+    model_name = repo_id.split("/")[-1]
     target_dir = os.path.join(save_dir, model_name)
     os.makedirs(target_dir, exist_ok=True)
     
     # Get file list from repo
     api = HfApi()
-    files = api.list_repo_files(repo_id.get('repo'), repo_type="model")
+    files = api.list_repo_files(repo_id, repo_type="model")
     
     for file in files:
         if not file.endswith('.png') and not file.endswith('.jpg') and not file.endswith('.webp'):
-            print(f"Downloading {file} from {repo_id.get('repo')}...")
-            hf_hub_download(repo_id=repo_id.get('repo'), filename=file, local_dir=target_dir)
+            print(f"Downloading {file} from {repo_id}...")
+            hf_hub_download(repo_id=repo_id, filename=file, local_dir=target_dir)
     
     return target_dir