| """Upload a finished run (metrics + final model) to a private HF model repo under runs/<run>/.""" |
|
|
| import argparse |
| from pathlib import Path |
|
|
| from huggingface_hub import HfApi |
|
|
|
|
| def main(): |
| p = argparse.ArgumentParser() |
| p.add_argument("--repo", default="Broyojo/world-pretraining-runs") |
| p.add_argument("--run", required=True) |
| p.add_argument("dirs", nargs="+", help="local dirs whose contents go to runs/<run>/") |
| args = p.parse_args() |
| api = HfApi() |
| api.create_repo(args.repo, private=True, exist_ok=True) |
| for d in args.dirs: |
| if Path(d).is_dir(): |
| api.upload_folder(repo_id=args.repo, folder_path=d, path_in_repo=f"runs/{args.run}", |
| commit_message=f"{args.run}: {Path(d).name}") |
| print(f"uploaded {d} -> {args.repo}/runs/{args.run}") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
