From 754e0442ae341d7f01c2c22ff8fb6a960aca96da Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=BD=90=E4=BF=9D=E5=85=83?= Date: Fri, 14 Nov 2025 19:02:10 +0800 Subject: [PATCH] [cpu] remove verbose print --- vllm_omni/worker/cpu/cpu_model_runner.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/vllm_omni/worker/cpu/cpu_model_runner.py b/vllm_omni/worker/cpu/cpu_model_runner.py index 7d532cc36..b90c39752 100644 --- a/vllm_omni/worker/cpu/cpu_model_runner.py +++ b/vllm_omni/worker/cpu/cpu_model_runner.py @@ -266,7 +266,6 @@ class OmniCPUModelRunner(CPUModelRunner): def extract_multimodal_outputs( self, hidden_states: Union[torch.Tensor, List[torch.Tensor]] ) -> dict: - print('hidden_states',hidden_states) if ( hasattr(self.model, "have_multimodal_outputs") and self.model.have_multimodal_outputs @@ -588,7 +587,7 @@ class OmniCPUModelRunner(CPUModelRunner): hidden_states, multimodal_outputs = self.extract_multimodal_outputs( hidden_states ) - if len(hidden_states.shape)==3: + if len(hidden_states.shape) == 3: return hidden_states[0], hidden_states[0][logit_indices] else: - return hidden_states, hidden_states[logit_indices] \ No newline at end of file + return hidden_states, hidden_states[logit_indices]