Fixed gpt-oss _load_weights_other() parameter position bug (#28715)
Co-authored-by: Dezhan Tu <dztu@meta.com>
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user