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:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user