Skip to content

Commit

Permalink
fix multi-images test prompts
Browse files Browse the repository at this point in the history
  • Loading branch information
Isotr0py committed Oct 21, 2024
1 parent 8d94421 commit 6e44174
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/models/decoder_only/vision_language/test_internvl.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
"cherry_blossom":
"<|im_start|>User\n<image>\nWhat is the season?<|im_end|>\n<|im_start|>Assistant\n", # noqa: E501
})
HF_MULTIIMAGE_IMAGE_PROMPT = "<|im_start|>User\nImage-1: <image>\nImage-2: <image>\nDescribe the two images in detail.<|im_end|>\n<|im_start|>Assistant\n" # noqa: E501
HF_MULTIIMAGE_IMAGE_PROMPT = "<|im_start|>User\nImage-1: <image>\nImage-2: <image>\nDescribe the two images in short.<|im_end|>\n<|im_start|>Assistant\n" # noqa: E501

models = [
"OpenGVLab/InternVL2-1B",
Expand Down

0 comments on commit 6e44174

Please sign in to comment.