Support file size == 0

tastelikefeet · Sep 10, 2024 · b3e3b68 · b3e3b68
1 parent 96e3387
commit b3e3b68
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/modelscope/hub/file_download.py b/modelscope/hub/file_download.py
@@ -591,14 +591,18 @@ 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,
             file_meta['Path'],
             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'])