mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-10 07:15:01 +08:00
Fix parameter order in GPT-OSS weight loading function for non-MXFP4 weights (#29506)
Signed-off-by: Filipp Fisin <48059208+qGentry@users.noreply.github.com> Co-authored-by: Cyrus Leung <tlleungac@connect.ust.hk>
This commit is contained in:
parent
b2c1d294fa
commit
5f5521bd5d
@ -647,8 +647,8 @@ class GptOssModel(nn.Module):
|
|||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
return self._load_weights_other(
|
return self._load_weights_other(
|
||||||
ep_rank_start,
|
|
||||||
ep_rank_end,
|
ep_rank_end,
|
||||||
|
ep_rank_start,
|
||||||
heads_per_rank,
|
heads_per_rank,
|
||||||
head_start,
|
head_start,
|
||||||
weights,
|
weights,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user