From b3e3b688af3abd646a3a9cd380e626b3fd15dcef Mon Sep 17 00:00:00 2001 From: "yuze.zyz" Date: Tue, 10 Sep 2024 16:18:32 +0800 Subject: [PATCH] Support file size == 0 --- modelscope/hub/file_download.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/modelscope/hub/file_download.py b/modelscope/hub/file_download.py index 7bbc49e1f..5dbc1ba52 100644 --- a/modelscope/hub/file_download.py +++ b/modelscope/hub/file_download.py @@ -591,7 +591,7 @@ def download_file(url, file_meta, temporary_cache_dir, cache, headers, headers=headers, cookies=None if cookies is None else cookies.get_dict(), file_size=file_meta['Size']) - else: + elif file_meta['Size'] > 0: http_get_model_file( url, temporary_cache_dir, @@ -599,6 +599,10 @@ def download_file(url, file_meta, temporary_cache_dir, cache, headers, file_size=file_meta['Size'], headers=headers, cookies=cookies) + else: + with open(os.path.join(temporary_cache_dir, file_meta['Path']), + 'w') as f: + f.write('') # check file integrity temp_file = os.path.join(temporary_cache_dir, file_meta['Path'])