xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-15 06:45:01 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats 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) 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: def sharded_weight_loader(shard_axis: int) -> LoaderFunction: