We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 221d388 commit fb30ee9Copy full SHA for fb30ee9
vllm/model_executor/models/blip2.py
@@ -481,14 +481,14 @@ def _get_prompt_replacements(
481
bos_token_id = tokenizer.bos_token_id
482
assert isinstance(bos_token_id, int)
483
484
- image_token_id = vocab["image"]
+ image_token_id = vocab["<image>"]
485
num_image_tokens = self.info.get_num_image_tokens()
486
image_tokens = [image_token_id] * num_image_tokens
487
488
return [
489
PromptReplacement(
490
modality="image",
491
- target="</s>",
+ target=[bos_token_id],
492
replacement=PromptReplacementDetails(
493
full=image_tokens + [bos_token_id],
494
features=image_tokens,
0 commit comments