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

2023-12-17 10:12:44 -08:00 · 2023-12-17 10:12:44 -08:00 · 2c9b638065
commit 2c9b638065
parent a7347d9a6d
1 changed files with 1 additions and 1 deletions
--- a/vllm/model_executor/weight_utils.py
+++ b/vllm/model_executor/weight_utils.py
@ -146,7 +146,7 @@ def prepare_hf_model_weights(
        raise ValueError(f"Unknown load_format: {load_format}")

    if fall_back_to_pt:
-        allow_patterns += [".pt"]
+        allow_patterns += ["*.pt"]

    if not is_local:
        # Use file lock to prevent multiple processes from