Spaces:
Running
Running
| """Publie le code (Space Docker) ou un checkpoint (modèle) sur le Hub via huggingface_hub. | |
| HF_TOKEN=... python scripts/push_to_hub.py --repo-id RDS777/mouh --kind space | |
| HF_TOKEN=... python scripts/push_to_hub.py --repo-id RDS777/mouh-1.5b --kind model --folder checkpoints/final | |
| python scripts/push_to_hub.py --repo-id x/y --kind space --dry-run # liste les fichiers sans rien envoyer | |
| """ | |
| import argparse | |
| import os | |
| import sys | |
| from pathlib import Path | |
| ROOT = Path(__file__).resolve().parents[1] | |
| IGNORE = ["__pycache__", "*.pyc", ".git", ".git/*", ".pytest_cache", "checkpoints", "checkpoints/*", "*.egg-info", ".env", "audit"] | |
| def main() -> None: | |
| ap = argparse.ArgumentParser(description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter) | |
| ap.add_argument("--repo-id", required=True) | |
| ap.add_argument("--kind", choices=["space", "model"], default="space") | |
| ap.add_argument("--folder", default=str(ROOT)) | |
| ap.add_argument("--private", action="store_true") | |
| ap.add_argument("--dry-run", action="store_true") | |
| ap.add_argument("--message", default="M.O.U.H. — mise à jour") | |
| a = ap.parse_args() | |
| folder = Path(a.folder) | |
| files = [p for p in folder.rglob("*") if p.is_file() and not any( | |
| part in {"__pycache__", ".git", ".pytest_cache", "checkpoints", "audit"} or part.endswith(".egg-info") | |
| for part in p.relative_to(folder).parts) and p.suffix != ".pyc" and p.name != ".env"] | |
| if a.dry_run: | |
| for p in sorted(files): | |
| print(p.relative_to(folder)) | |
| print(f"{len(files)} fichier(s) seraient envoyés vers {a.kind} {a.repo_id}") | |
| return | |
| token = os.environ.get("HF_TOKEN") | |
| if not token: | |
| sys.exit("HF_TOKEN absent : définissez un jeton avec droit d'écriture") | |
| from huggingface_hub import HfApi | |
| api = HfApi(token=token) | |
| if a.kind == "space": | |
| api.create_repo(a.repo_id, repo_type="space", space_sdk="docker", exist_ok=True, private=a.private) | |
| else: | |
| api.create_repo(a.repo_id, repo_type="model", exist_ok=True, private=a.private) | |
| info = api.upload_folder(folder_path=str(folder), repo_id=a.repo_id, repo_type=a.kind, | |
| ignore_patterns=IGNORE, commit_message=a.message) | |
| print(info) | |
| if __name__ == "__main__": | |
| main() | |