Skip to content

Commit fb30ee9

Browse files
[Bugfix] Fix BLIP-2 processing (#12412)
Signed-off-by: DarkLight1337 <[email protected]>
1 parent 221d388 commit fb30ee9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/model_executor/models/blip2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -481,14 +481,14 @@ def _get_prompt_replacements(
481481
bos_token_id = tokenizer.bos_token_id
482482
assert isinstance(bos_token_id, int)
483483

484-
image_token_id = vocab["image"]
484+
image_token_id = vocab["<image>"]
485485
num_image_tokens = self.info.get_num_image_tokens()
486486
image_tokens = [image_token_id] * num_image_tokens
487487

488488
return [
489489
PromptReplacement(
490490
modality="image",
491-
target="</s>",
491+
target=[bos_token_id],
492492
replacement=PromptReplacementDetails(
493493
full=image_tokens + [bos_token_id],
494494
features=image_tokens,

0 commit comments

Comments
 (0)