Skip to content

Commit 6c6dcd8

Browse files
authored
[MISC] correct signature for LoaderFunction (#18670)
Signed-off-by: Andy Xie <andy.xning@gmail.com>
1 parent 7891fdf commit 6c6dcd8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/model_executor/model_loader/weight_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -634,7 +634,7 @@ def row_parallel_weight_loader(param: torch.Tensor,
634634
return default_weight_loader(param, loaded_weight)
635635

636636

637-
LoaderFunction = Callable[[torch.Tensor, torch.Tensor], torch.Tensor]
637+
LoaderFunction = Callable[[torch.Tensor, torch.Tensor], None]
638638

639639

640640
def sharded_weight_loader(shard_axis: int) -> LoaderFunction:

0 commit comments

Comments
 (0)