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

fix log for downloading to local dir

Browse Source
This commit is contained in:
Yingda Chen
2024-11-12 09:15:50 +08:00
parent da47da41bc
commit b190fc96b0
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modelscope/hub/snapshot_download.py
View File

@@ -221,7 +221,9 @@ def _snapshot_download(
cookies = ModelScopeConfig.get_cookies()
repo_files = []
if repo_type == REPO_TYPE_MODEL:
directory = os.path.join(system_cache, 'hub', repo_id)
directory = os.path.abspath(
local_dir) if local_dir is not None else os.path.join(
system_cache, 'hub', repo_id)
print(f'Downloading Model to directory: {directory}')
revision_detail = _api.get_valid_revision_detail(
repo_id, revision=revision, cookies=cookies)
@@ -274,7 +276,9 @@ def _snapshot_download(
)
elif repo_type == REPO_TYPE_DATASET:
directory = os.path.join(system_cache, 'datasets', repo_id)
directory = os.path.abspath(
local_dir) if local_dir is not None else os.path.join(
system_cache, 'datasets', repo_id)
print(f'Downloading Dataset to directory: {directory}')
group_or_owner, name = model_id_to_group_owner_name(repo_id)
if not revision: