adapt MODELSCOPE download (#17922)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user