biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Model] Remove redundant None check in DeepSeekOCR image input processing (#32016)

Browse Source 
Signed-off-by: maang <maang_h@163.com>
This commit is contained in:
maang
2026-01-09 22:12:44 +08:00
committed by GitHub
parent bbf80ede43
commit 7cdf7e2fe0
1 changed files with 10 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
vllm/model_executor/models/deepseek_ocr.py
View File

@@ -438,7 +438,6 @@ 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",
@@ -450,8 +449,6 @@ class DeepseekOCRForCausalLM(nn.Module, SupportsMultiModal, SupportsPP, Supports
},
)
raise AssertionError("This line should be unreachable.")
def _encode_global_features(self, image_tensor: torch.Tensor) -> torch.Tensor:
global_features_1 = self.sam_model(image_tensor)
global_features_2 = self.vision_model(image_tensor, global_features_1)