From 33bdd0d21519637fa2eaa379b439ee6192d9e8a4 Mon Sep 17 00:00:00 2001 From: "mulin.lyh" Date: Fri, 26 Jul 2024 16:34:47 +0800 Subject: [PATCH] fix bug --- modelscope/hub/snapshot_download.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/modelscope/hub/snapshot_download.py b/modelscope/hub/snapshot_download.py index cda9f934..7f069795 100644 --- a/modelscope/hub/snapshot_download.py +++ b/modelscope/hub/snapshot_download.py @@ -255,6 +255,7 @@ def _snapshot_download( cookies=cookies, ignore_file_pattern=ignore_file_pattern, allow_file_pattern=allow_file_pattern) + elif repo_type == REPO_TYPE_DATASET: group_or_owner, name = model_id_to_group_owner_name(repo_id) if not revision: @@ -299,6 +300,9 @@ def _snapshot_download( break page_number += 1 + cache.save_model_version(revision_info=revision_detail) + return os.path.join(cache.get_root_location()) + def _download_file_lists( repo_files: List[str], @@ -372,6 +376,3 @@ def _download_file_lists( download_file(url, repo_file, temporary_cache_dir, cache, headers, cookies) - - cache.save_model_version(revision_info=revision_detail) - return os.path.join(cache.get_root_location())