Skip to content

GitLab

Unverified Commit 7cdf7e2f authored by maang's avatar maang Committed by GitHub
Browse files

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


Signed-off-by: default avatarmaang <maang_h@163.com>
parent bbf80ede
Show whitespace changes
Inline Side-by-side
Showing with 10 additions and 13 deletions
vllm/model_executor/models/deepseek_ocr.py
View file @ 7cdf7e2f
...@@ -438,7 +438,6 @@ class DeepseekOCRForCausalLM(nn.Module, SupportsMultiModal, SupportsPP, Supports ...@@ -438,7 +438,6 @@ class DeepseekOCRForCausalLM(nn.Module, SupportsMultiModal, SupportsPP, Supports
if pixel_values is None or torch.sum(pixel_values).item() == 0: if pixel_values is None or torch.sum(pixel_values).item() == 0:
return None return None
if pixel_values is not None:
base_size = self.vision_config.image_size base_size = self.vision_config.image_size
return DeepseekOCRImagePixelInputs( return DeepseekOCRImagePixelInputs(
type="pixel_values", type="pixel_values",
...@@ -450,8 +449,6 @@ class DeepseekOCRForCausalLM(nn.Module, SupportsMultiModal, SupportsPP, Supports ...@@ -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: def _encode_global_features(self, image_tensor: torch.Tensor) -> torch.Tensor:
global_features_1 = self.sam_model(image_tensor) global_features_1 = self.sam_model(image_tensor)
global_features_2 = self.vision_model(image_tensor, global_features_1) global_features_2 = self.vision_model(image_tensor, global_features_1)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment