From 2c9b638065ae8d69f8949e56f87990f1fcfe9255 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Sun, 17 Dec 2023 10:12:44 -0800 Subject: [PATCH] [Minor] Fix a typo in .pt weight support (#2160) --- vllm/model_executor/weight_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/weight_utils.py b/vllm/model_executor/weight_utils.py index bff4fb2f..365c847a 100644 --- 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