frozenleaves/vllm
1
0
Fork 0
mirror of https://github.com/vllm-project/vllm.git synced 2025-10-20 14:53:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

[MISC] correct signature for LoaderFunction (#18670)

Browse Source 
Signed-off-by: Andy Xie <andy.xning@gmail.com>
This commit is contained in:
Ning Xie
2025-05-25 11:17:47 +08:00
committed by GitHub
parent 7891fdf0c6
commit 6c6dcd8611
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/model_executor/model_loader/weight_utils.py
View File

@ -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: