Unverified Commit 5f5521bd authored by Filipp Fisin's avatar Filipp Fisin Committed by GitHub
Browse files

Fix parameter order in GPT-OSS weight loading function for non-MXFP4 weights (#29506)


Signed-off-by: default avatarFilipp Fisin <48059208+qGentry@users.noreply.github.com>
Co-authored-by: default avatarCyrus Leung <tlleungac@connect.ust.hk>
parent b2c1d294
......@@ -647,8 +647,8 @@ class GptOssModel(nn.Module):
)
else:
return self._load_weights_other(
ep_rank_start,
ep_rank_end,
ep_rank_start,
heads_per_rank,
head_start,
weights,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment