From 7cdf7e2fe0d406724659d84e0a17fde97f855a89 Mon Sep 17 00:00:00 2001 From: maang Date: Fri, 9 Jan 2026 22:12:44 +0800 Subject: [PATCH] [Model] Remove redundant None check in DeepSeekOCR image input processing (#32016) Signed-off-by: maang --- vllm/model_executor/models/deepseek_ocr.py | 23 ++++++++++------------ 1 file changed, 10 insertions(+), 13 deletions(-) diff --git a/vllm/model_executor/models/deepseek_ocr.py b/vllm/model_executor/models/deepseek_ocr.py index 87afec0d3..f396897bf 100644 --- a/vllm/model_executor/models/deepseek_ocr.py +++ b/vllm/model_executor/models/deepseek_ocr.py @@ -438,19 +438,16 @@ class DeepseekOCRForCausalLM(nn.Module, SupportsMultiModal, SupportsPP, Supports if pixel_values is None or torch.sum(pixel_values).item() == 0: return None - if pixel_values is not None: - base_size = self.vision_config.image_size - return DeepseekOCRImagePixelInputs( - type="pixel_values", - data=pixel_values, - images_crop=images_crop, - images_spatial_crop=images_spatial_crop, - resolve_bindings={ - "base_size": base_size, - }, - ) - - raise AssertionError("This line should be unreachable.") + base_size = self.vision_config.image_size + return DeepseekOCRImagePixelInputs( + type="pixel_values", + data=pixel_values, + images_crop=images_crop, + images_spatial_crop=images_spatial_crop, + resolve_bindings={ + "base_size": base_size, + }, + ) def _encode_global_features(self, image_tensor: torch.Tensor) -> torch.Tensor: global_features_1 = self.sam_model(image_tensor)