mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-16 04:55:02 +08:00
Fixed gpt-oss _load_weights_other() parameter position bug (#28715)
Co-authored-by: Dezhan Tu <dztu@meta.com>
This commit is contained in:
parent
b316ac6589
commit
af02c40970
@ -641,8 +641,8 @@ class GptOssModel(nn.Module):
|
|||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
return self._load_weights_other(
|
return self._load_weights_other(
|
||||||
ep_rank_end,
|
|
||||||
ep_rank_start,
|
ep_rank_start,
|
||||||
|
ep_rank_end,
|
||||||
heads_per_rank,
|
heads_per_rank,
|
||||||
head_start,
|
head_start,
|
||||||
weights,
|
weights,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user