|
|
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, |
|
|
) |
|
|
|
