diff --git a/vllm/model_executor/models/gpt_oss.py b/vllm/model_executor/models/gpt_oss.py index 9de3e261941b..cff16b7a7a8c 100644 --- a/vllm/model_executor/models/gpt_oss.py +++ b/vllm/model_executor/models/gpt_oss.py @@ -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,