Skip to content

GitLab

Unverified Commit 04d0c607 authored by damon's avatar damon Committed by GitHub
Browse files

[Bugfix] Fix the issue that Blip2ForConditionalGeneration' object has… (#24028) 


Signed-off-by: default avatarDazhi Jiang <dazhi_jiang@163.com>
parent 2b41cbbf
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
vllm/model_executor/models/blip2.py
View file @ 04d0c607
...@@ -560,8 +560,8 @@ class Blip2ForConditionalGeneration(nn.Module, SupportsMultiModal, SupportsPP, ...@@ -560,8 +560,8 @@ class Blip2ForConditionalGeneration(nn.Module, SupportsMultiModal, SupportsPP,
self.make_empty_intermediate_tensors = ( self.make_empty_intermediate_tensors = (
self.language_model.make_empty_intermediate_tensors) self.language_model.make_empty_intermediate_tensors)
def _create_image_input(self, def _parse_and_validate_image_input(
**kwargs: object) -> Optional[Blip2ImageInputs]: self, **kwargs: object) -> Optional[Blip2ImageInputs]:
pixel_values = kwargs.pop("pixel_values", None) pixel_values = kwargs.pop("pixel_values", None)
image_embeds = kwargs.pop("image_embeds", None) image_embeds = kwargs.pop("image_embeds", None)
......
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