[Model] [gpt-oss] fix gpt-oss pp support (#23815)

Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
2025-08-28 20:36:55 +08:00
parent a3432f18fd
commit bfab219648
1 changed files with 2 additions and 3 deletions
--- a/vllm/model_executor/models/gpt_oss.py
+++ b/vllm/model_executor/models/gpt_oss.py
@ -668,9 +668,8 @@ class GptOssForCausalLM(nn.Module, SupportsPP):
                positions: torch.Tensor,
                intermediate_tensors: Optional[IntermediateTensors] = None,
                inputs_embeds: Optional[torch.Tensor] = None) -> torch.Tensor:
-        assert intermediate_tensors is None
-        assert inputs_embeds is None
-        return self.model(input_ids, positions)
+        return self.model(input_ids, positions, intermediate_tensors,
+                          inputs_embeds)

    def compute_logits(self, hidden_states: torch.Tensor,
                       sampling_metadata: SamplingMetadata) -> torch.Tensor: