Skip to content

Commit 19df881

Browse files
ahao-anyscalecharlifu
authored andcommitted
[BUG] Exclude .pth files when pulling remote files (vllm-project#25092)
Signed-off-by: ahao-anyscale <ahao@anyscale.com> Signed-off-by: charlifu <charlifu@amd.com>
1 parent 55aacf3 commit 19df881

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

vllm/config/__init__.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -845,17 +845,21 @@ def maybe_pull_model_tokenizer_for_runai(self, model: str,
845845
object_storage_model.pull_files(model,
846846
ignore_pattern=[
847847
"*.pt", "*.safetensors",
848-
"*.bin", "*.tensors"
848+
"*.bin", "*.tensors",
849+
"*.pth"
849850
])
850851
self.tokenizer = object_storage_model.dir
851852
return
852853

853854
# Only download tokenizer if needed and not already handled
854855
if is_runai_obj_uri(tokenizer):
855856
object_storage_tokenizer = ObjectStorageModel()
856-
object_storage_tokenizer.pull_files(
857-
model,
858-
ignore_pattern=["*.pt", "*.safetensors", "*.bin", "*.tensors"])
857+
object_storage_tokenizer.pull_files(model,
858+
ignore_pattern=[
859+
"*.pt", "*.safetensors",
860+
"*.bin", "*.tensors",
861+
"*.pth"
862+
])
859863
self.tokenizer = object_storage_tokenizer.dir
860864

861865
def _get_encoder_config(self):

0 commit comments

Comments
 (0)