Fixed gpt-oss _load_weights_other() parameter position bug (#28715)

Co-authored-by: Dezhan Tu <dztu@meta.com>
This commit is contained in:
Dezhan 2025-11-16 01:46:29 -08:00 committed by GitHub
parent b316ac6589
commit af02c40970
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -641,8 +641,8 @@ class GptOssModel(nn.Module):
)
else:
return self._load_weights_other(
ep_rank_end,
ep_rank_start,
ep_rank_end,
heads_per_rank,
head_start,
weights,