初始化项目,由ModelHub XC社区提供模型
Model: ksjpswaroop/zindango-slm Source: Original Platform
This commit is contained in:
63
scripts/push_scripts_to_hub.py
Normal file
63
scripts/push_scripts_to_hub.py
Normal file
@@ -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()
|
||||
Reference in New Issue
Block a user