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

Signed-off-by: Dazhi Jiang <dazhi_jiang@163.com>
This commit is contained in:
damon 2025-09-02 11:54:20 +08:00 committed by GitHub
parent 2b41cbbf03
commit 04d0c60770
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -560,8 +560,8 @@ class Blip2ForConditionalGeneration(nn.Module, SupportsMultiModal, SupportsPP,
self.make_empty_intermediate_tensors = (
self.language_model.make_empty_intermediate_tensors)
def _create_image_input(self,
**kwargs: object) -> Optional[Blip2ImageInputs]:
def _parse_and_validate_image_input(
self, **kwargs: object) -> Optional[Blip2ImageInputs]:
pixel_values = kwargs.pop("pixel_values", None)
image_embeds = kwargs.pop("image_embeds", None)