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): ...@@ -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,
......
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