File size: 2,127 Bytes
e071965 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | #!/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()
|