squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Minor] Fix a typo in .pt weight support (#2160)

Browse Source
This commit is contained in:
Woosuk Kwon 2023-12-17 10:12:44 -08:00 committed by GitHub
parent a7347d9a6d
commit 2c9b638065
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/model_executor/weight_utils.py
View File

@ -146,7 +146,7 @@ def prepare_hf_model_weights(
raise ValueError(f"Unknown load_format: {load_format}") raise ValueError(f"Unknown load_format: {load_format}")
if fall_back_to_pt: if fall_back_to_pt:
allow_patterns += [".pt"] allow_patterns += ["*.pt"]
if not is_local: if not is_local:
# Use file lock to prevent multiple processes from # Use file lock to prevent multiple processes from