File size: 794 Bytes
32bfe02 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | """Upload weights to ModelScope. Usage: MS_TOKEN=xxx python scripts/push_ms.py [namespace/g2pS]"""
import os
import sys
from modelscope.hub.api import HubApi
repo = sys.argv[1] if len(sys.argv) > 1 else 'baicai-1145/g2pS'
api = HubApi()
api.login(os.environ['MS_TOKEN'])
api.create_repo(repo, repo_type='model', exist_ok=True)
# upload weights + meta (skip code dirs)
folders = ['models', 'torch_weights']
files = ['README.md', 'MODEL_CARD.md', 'LICENSE']
for f in files:
api.upload_file(repo_id=repo, repo_type='model', path_or_fileobj=f, path_in_repo=f)
for d in folders:
for root, _, fs in os.walk(d):
for f in fs:
p = os.path.join(root, f)
api.upload_file(repo_id=repo, repo_type='model', path_or_fileobj=p, path_in_repo=p)
print('uploaded', repo)
|