diff --git a/vllm/worker/cpu_model_runner.py b/vllm/worker/cpu_model_runner.py index b7002e75c9ef5..d7d7d65659b73 100644 --- a/vllm/worker/cpu_model_runner.py +++ b/vllm/worker/cpu_model_runner.py @@ -456,6 +456,8 @@ def execute_model( model_input.attn_metadata, **MultiModalInputs.as_kwargs(model_input.multi_modal_kwargs or {}, device=self.device), + "intermediate_tensors": + intermediate_tensors, } hidden_states = model_executable(**execute_model_kwargs)