adapt MODELSCOPE download (#17922)

This commit is contained in:
jianzhao-xu
2026-01-30 15:26:54 +08:00
committed by GitHub
parent 70db3398d1
commit 96584ab692

View File

@@ -850,9 +850,14 @@ class ServerArgs:
if not os.path.exists(self.model_path):
from modelscope import snapshot_download
self.model_path = snapshot_download(self.model_path)
self.model_path = snapshot_download(
self.model_path, cache_dir=self.download_dir, revision=self.revision
)
self.tokenizer_path = snapshot_download(
self.tokenizer_path, ignore_patterns=["*.bin", "*.safetensors"]
self.tokenizer_path,
cache_dir=self.download_dir,
revision=self.revision,
ignore_patterns=["*.bin", "*.safetensors"],
)
# Mamba scheduler strategy