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:
Filipp Fisin 2025-11-28 09:45:10 +01:00 committed by GitHub
parent b2c1d294fa
commit 5f5521bd5d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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,