Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 04:01:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix hub push model token (#1592)

Browse Source
This commit is contained in:
Yunlin Mao
2026-01-15 15:11:34 +08:00
committed by GitHub
parent e235f3f26e
commit bb07c16abf
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modelscope/hub/api.py
View File

@@ -856,20 +856,21 @@ class HubApi:
ignore_file_pattern = [ignore_file_pattern]
if visibility is None or license is None:
raise InvalidParameter('Visibility and License cannot be empty for new model.')
if not self.repo_exists(model_id):
if not self.repo_exists(model_id, token=token):
logger.info('Creating new model [%s]' % model_id)
self.create_model(
model_id=model_id,
visibility=visibility,
license=license,
chinese_name=chinese_name,
original_model_id=original_model_id)
original_model_id=original_model_id,
token=token)
tmp_dir = os.path.join(model_dir, TEMPORARY_FOLDER_NAME) # make temporary folder
git_wrapper = GitCommandWrapper()
logger.info(f'Pushing folder {model_dir} as model {model_id}.')
logger.info(f'Total folder size {folder_size}, this may take a while depending on actual pushing size...')
try:
repo = Repository(model_dir=tmp_dir, clone_from=model_id)
repo = Repository(model_dir=tmp_dir, clone_from=model_id, auth_token=token)
branches = git_wrapper.get_remote_branches(tmp_dir)
if revision not in branches:
logger.info('Creating new branch %s' % revision)
@@ -2050,7 +2051,7 @@ class HubApi:
if create_default_config:
with tempfile.TemporaryDirectory() as temp_cache_dir:
from modelscope.hub.repository import Repository
repo = Repository(temp_cache_dir, repo_id)
repo = Repository(temp_cache_dir, repo_id, auth_token=token)
default_config = {
'framework': 'pytorch',
'task': 'text-generation',