xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 08:04:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix] fix modelscope snapshot_download serialization (#21536)

Browse Source 
Signed-off-by: Andy Xie <andy.xning@gmail.com>
This commit is contained in:
Ning Xie 2025-07-25 13:44:38 +08:00 committed by GitHub
parent 42343f1f89
commit adaf2c6d4f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
vllm/model_executor/model_loader/default_loader.py
View File

@ -69,10 +69,10 @@ class DefaultModelLoader(BaseModelLoader):
# pylint: disable=C.
from modelscope.hub.snapshot_download import snapshot_download
if not os.path.exists(model):
# Use file lock to prevent multiple processes from
# downloading the same model weights at the same time.
with get_lock(model, self.load_config.download_dir):
# Use file lock to prevent multiple processes from
# downloading the same model weights at the same time.
with get_lock(model, self.load_config.download_dir):
if not os.path.exists(model):
model_path = snapshot_download(
model_id=model,
cache_dir=self.load_config.download_dir,
@ -81,8 +81,8 @@ class DefaultModelLoader(BaseModelLoader):
revision=revision,
ignore_file_pattern=self.load_config.ignore_patterns,
)
else:
model_path = model
else:
model_path = model
return model_path
return None