| """Check the release and, with --yes, upload it to the Hugging Face Hub. |
| |
| Usage: python scripts/publish.py (checks only, uploads nothing) |
| python scripts/publish.py --yes (creates the public repo and uploads) |
| Requires `pip install huggingface_hub` and `hf auth login`. |
| """ |
| import argparse |
| import json |
| import sys |
| from pathlib import Path |
|
|
| ROOT = Path(__file__).resolve().parent.parent |
| IGNORE = ["_private/*", ".git/*", ".gitignore", "card/*", "**/__pycache__/*"] |
| REQUIRED = ["README.md", "LICENSE", "NOTICE", "SHA256SUMS", "Modelfile", "template", "params", |
| "Sagui-4B-IQ3_XXS.gguf", "mmproj-Sagui-4B-F16.gguf", "assets/banner.png"] |
|
|
|
|
| def problems(cfg): |
| found = [f"missing file: {name}" for name in REQUIRED if not (ROOT / name).exists()] |
| if not any(cfg["donations"].values()): |
| found.append("release.json: no donation link filled in") |
| readme = (ROOT / "README.md").read_text(encoding="utf-8") if (ROOT / "README.md").exists() else "" |
| if "@@" in readme: |
| found.append("README.md has unrendered placeholders: run scripts/render_card.py") |
| return found |
|
|
|
|
| def main(): |
| ap = argparse.ArgumentParser() |
| ap.add_argument("--yes", action="store_true", help="really create the public repo and upload") |
| args = ap.parse_args() |
| cfg = json.loads((ROOT / "release.json").read_text(encoding="utf-8")) |
| repo_id = f"{cfg['hf_namespace']}/{cfg['repo_name']}" |
|
|
| found = problems(cfg) |
| for problem in found: |
| print("BLOCKED:", problem) |
| if found: |
| sys.exit(1) |
| print(f"OK: ready to publish {repo_id}") |
| if not args.yes: |
| print("Nothing uploaded. Run again with --yes to publish.") |
| return |
|
|
| from huggingface_hub import HfApi |
| api = HfApi() |
| api.create_repo(repo_id, repo_type="model", exist_ok=True) |
| api.upload_folder(repo_id=repo_id, folder_path=ROOT, ignore_patterns=IGNORE, |
| commit_message="Release Sagui-4B") |
| print(f"https://huggingface.co/{repo_id}") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
