diff --git a/modelscope/hub/api.py b/modelscope/hub/api.py index 999f1336..10747f85 100644 --- a/modelscope/hub/api.py +++ b/modelscope/hub/api.py @@ -8,7 +8,6 @@ import pickle import platform import re import shutil -import tempfile import uuid from collections import defaultdict from http import HTTPStatus @@ -34,8 +33,8 @@ from modelscope.hub.constants import (API_HTTP_CLIENT_MAX_RETRIES, MODELSCOPE_CLOUD_USERNAME, MODELSCOPE_REQUEST_ID, ONE_YEAR_SECONDS, REQUESTS_API_HTTP_METHOD, - DatasetVisibility, Licenses, - ModelVisibility) + TEMPORARY_FOLDER_NAME, DatasetVisibility, + Licenses, ModelVisibility) from modelscope.hub.errors import (InvalidParameter, NotExistError, NotLoginException, NoValidRevisionError, RequestError, datahub_raise_on_error, @@ -391,7 +390,7 @@ class HubApi: license=license, chinese_name=chinese_name, original_model_id=original_model_id) - tmp_dir = tempfile.mkdtemp() + 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...') @@ -433,6 +432,7 @@ class HubApi: remote_branch=revision) if tag is not None: repo.tag_and_push(tag, tag) + logger.info(f'Successfully push folder {model_dir} to remote {model_id}.') except Exception: raise finally: