Upload upload_lora_to_hf.py with huggingface_hub

upload_lora_to_hf.py

@@ -8,8 +8,8 @@ from huggingface_hub import HfApi
 def parse_args():
     parser = argparse.ArgumentParser(description="Upload a LoRA adapter folder to a Hugging Face model repo")
     parser.add_argument("--repo-id", default="ASTERIZER/LUNA-100M")
-    parser.add_argument("--folder", default="Base/out/sft/rag_mcp_lora/final")
-    parser.add_argument("--path-in-repo", default="rag_mcp_lora/final")
+    parser.add_argument("--folder", default="Base/out/sft/rag_mcp_lora")
+    parser.add_argument("--path-in-repo", default="rag_mcp_lora")
     parser.add_argument("--private", action="store_true")
     return parser.parse_args()
 
@@ -26,9 +26,15 @@ def main():
     if not folder.is_dir():
         raise NotADirectoryError(f"Expected a folder, got: {folder}")
 
-    required_files = [folder / "adapter_model.pt", folder / "adapter_bundle.pt"]
-    missing = [str(path) for path in required_files if not path.exists()]
-    if missing:
+    candidate_files = [
+        folder / "adapter_model.pt",
+        folder / "adapter_bundle.pt",
+        folder / "final" / "adapter_model.pt",
+        folder / "final" / "adapter_bundle.pt",
+    ]
+    existing = [path for path in candidate_files if path.exists()]
+    missing = [str(path) for path in candidate_files if not path.exists()]
+    if not existing:
         raise FileNotFoundError(f"Missing expected adapter files: {', '.join(missing)}")
 
     api = HfApi(token=token)