diff --git a/modelscope/hub/snapshot_download.py b/modelscope/hub/snapshot_download.py index 35b0f3a4..77b49847 100644 --- a/modelscope/hub/snapshot_download.py +++ b/modelscope/hub/snapshot_download.py @@ -20,7 +20,7 @@ from modelscope.utils.constant import (DEFAULT_DATASET_REVISION, DEFAULT_REPOSITORY_REVISION, REPO_TYPE_DATASET, REPO_TYPE_MODEL, REPO_TYPE_SUPPORT) -from modelscope.utils.file_utils import get_default_modelscope_cache_dir +from modelscope.utils.file_utils import get_modelscope_cache_dir from modelscope.utils.logger import get_logger from modelscope.utils.thread_utils import thread_executor @@ -222,7 +222,7 @@ def _snapshot_download( temporary_cache_dir, cache = create_temporary_directory_and_cache( repo_id, local_dir=local_dir, cache_dir=cache_dir, repo_type=repo_type) - system_cache = cache_dir if cache_dir is not None else get_default_modelscope_cache_dir( + system_cache = cache_dir if cache_dir is not None else get_modelscope_cache_dir( ) if local_files_only: if len(cache.cached_files) == 0: