From 6c6dcd8611933fb2dd1842b26e63c06a9b018e4f Mon Sep 17 00:00:00 2001 From: Ning Xie Date: Sun, 25 May 2025 11:17:47 +0800 Subject: [PATCH] [MISC] correct signature for LoaderFunction (#18670) Signed-off-by: Andy Xie --- vllm/model_executor/model_loader/weight_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/model_loader/weight_utils.py b/vllm/model_executor/model_loader/weight_utils.py index a1cf43328bab1..1547a016ab883 100644 --- a/vllm/model_executor/model_loader/weight_utils.py +++ b/vllm/model_executor/model_loader/weight_utils.py @@ -634,7 +634,7 @@ def row_parallel_weight_loader(param: torch.Tensor, return default_weight_loader(param, loaded_weight) -LoaderFunction = Callable[[torch.Tensor, torch.Tensor], torch.Tensor] +LoaderFunction = Callable[[torch.Tensor, torch.Tensor], None] def sharded_weight_loader(shard_axis: int) -> LoaderFunction: