| import argparse |
| from huggingface_hub import HfApi |
|
|
| parser = argparse.ArgumentParser() |
| parser.add_argument('--public_repo', action='store_false', help='makes repo public') |
| parser.add_argument('--folder_path', type=str) |
| parser.add_argument('--repo_id', type=str) |
| parser.add_argument('--no_multi_commits', action='store_false', help='in chunks for larger uploads') |
| parser.add_argument('--write_token', type=str) |
|
|
| ''' |
| If fails, just rerun. See https://huggingface.co/docs/huggingface_hub/guides/upload#upload-a-folder-by-chunks |
| ''' |
|
|
| if __name__ == "__main__": |
| api = HfApi() |
| args = parser.parse_args() |
| api.create_repo( |
| repo_id=args.repo_id, |
| token=args.write_token, |
| private=args.public_repo, |
| repo_type="model", |
| exist_ok=True, |
| ) |
| api.upload_folder( |
| folder_path=args.folder_path, |
| repo_id=args.repo_id, |
| repo_type="model", |
| multi_commits=args.no_multi_commits, |
| multi_commits_verbose=True, |
| token=args.write_token, |
| create_pr=True, |
| ) |
|
|
