Add scripts/push_scripts_to_hub.py

scripts/push_scripts_to_hub.py

@@ -0,0 +1,63 @@
+#!/usr/bin/env python3
+"""
+Push zindango-slm scripts to Hugging Face model repo (scripts/ folder).
+
+Usage:
+  python scripts/push_scripts_to_hub.py [--repo-id USERNAME/zindango-slm]
+"""
+
+import argparse
+from pathlib import Path
+
+from huggingface_hub import HfApi, create_repo, upload_file
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--repo-id", default=None)
+    parser.add_argument("--skip-create", action="store_true")
+    args = parser.parse_args()
+
+    sft_root = Path(__file__).resolve().parent.parent
+    benchmark_root = sft_root.parent / "nanbeige-benchmark-verification"
+
+    api = HfApi()
+    user = api.whoami()
+    username = user["name"]
+    repo_id = args.repo_id or f"{username}/zindango-slm"
+
+    if not args.skip_create:
+        try:
+            create_repo(repo_id, repo_type="model", exist_ok=True)
+        except Exception as e:
+            if "403" in str(e).lower() or "forbidden" in str(e).lower():
+                print("Create repo failed. Run with --skip-create.")
+                raise
+
+    scripts_to_upload = [
+        (sft_root / "scripts/convert_and_push_gguf.py", "scripts/convert_and_push_gguf.py"),
+        (sft_root / "scripts/push_to_hub.py", "scripts/push_to_hub.py"),
+        (sft_root / "scripts/push_scripts_to_hub.py", "scripts/push_scripts_to_hub.py"),
+        (benchmark_root / "scripts/test_zindango_gguf.py", "scripts/test_zindango_gguf.py"),
+        (benchmark_root / "scripts/llamacpp_chat.py", "scripts/llamacpp_chat.py"),
+        (benchmark_root / "scripts/llamacpp_chat.sh", "scripts/llamacpp_chat.sh"),
+    ]
+
+    for local_path, path_in_repo in scripts_to_upload:
+        if not local_path.exists():
+            print(f"Skipping {local_path} (not found)")
+            continue
+        print(f"Uploading {path_in_repo}...")
+        upload_file(
+            path_or_fileobj=str(local_path),
+            path_in_repo=path_in_repo,
+            repo_id=repo_id,
+            repo_type="model",
+            commit_message=f"Add {path_in_repo}",
+        )
+
+    print(f"Done. https://huggingface.co/{repo_id}")
+
+
+if __name__ == "__main__":
+    main()