diff --git a/vllm/model_executor/models/blip2.py b/vllm/model_executor/models/blip2.py index 09c5087c2dc36..b559ac677a740 100644 --- a/vllm/model_executor/models/blip2.py +++ b/vllm/model_executor/models/blip2.py @@ -481,14 +481,14 @@ def _get_prompt_replacements( bos_token_id = tokenizer.bos_token_id assert isinstance(bos_token_id, int) - image_token_id = vocab["image"] + image_token_id = vocab[""] num_image_tokens = self.info.get_num_image_tokens() image_tokens = [image_token_id] * num_image_tokens return [ PromptReplacement( modality="image", - target="", + target=[bos_token_id], replacement=PromptReplacementDetails( full=image_tokens + [bos_token_id], features=image_tokens,