lucas
/
LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

try to past test

This commit is contained in:
BUAADreamer 2024-09-10 13:29:09 +08:00
parent e387216d49
commit 0042a144d5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/data/test_mm_plugin.py
View File

@ -142,7 +142,7 @@ def test_idefics2_plugin():
check_inputs = {"plugin": idefics2_plugin, "tokenizer": tokenizer, "processor": processor}
mm_messages = copy.deepcopy(MM_MESSAGES)
fake_image_token = processor.fake_image_token.content
image_str = f"{fake_image_token}{"<image>" * processor.image_seq_len}{fake_image_token}"
image_str = f"{fake_image_token}{'<image>' * processor.image_seq_len}{fake_image_token}"
image_str = image_str * 5
for message in mm_messages:
content = message["content"]