diff --git a/vllm/model_executor/models/step3_text.py b/vllm/model_executor/models/step3_text.py index a0e7e16a9bb..dd4af6f0fec 100644 --- a/vllm/model_executor/models/step3_text.py +++ b/vllm/model_executor/models/step3_text.py @@ -345,7 +345,7 @@ class Step3TextModel(nn.Module): self.norm = PPMissingLayer() self.make_empty_intermediate_tensors = make_empty_intermediate_tensors_factory( - ["hidden_states"], config.hidden_size + ["hidden_states", "residual"], config.hidden_size ) def embed_input_ids(self, input_ids: torch.Tensor) -> torch.Tensor: