| """Upload weights to HF. Usage: HF_TOKEN=hf_xxx python scripts/push_hf.py""" | |
| import os, sys | |
| from huggingface_hub import HfApi | |
| repo = sys.argv[1] if len(sys.argv) > 1 else 'baicai-1145/g2pS' | |
| api = HfApi(token=os.environ['HF_TOKEN']) | |
| api.create_repo(repo, repo_type='model', exist_ok=True) | |
| api.upload_large_folder(repo_id=repo, repo_type='model', folder_path='.', | |
| ignore_patterns=['.git/*', 'training/*', 'scripts/*', 'examples/*', 'g2ps/*']) | |
| print('uploaded', repo) | |