423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 23:58:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

try to past test

Browse Source 
Former-commit-id: 2db97e1e5e06370375f4f5c577671524e399321f
This commit is contained in:
BUAADreamer 2024-09-10 13:29:09 +08:00
parent df24345d65
commit ec696bbcdd
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} check_inputs = {"plugin": idefics2_plugin, "tokenizer": tokenizer, "processor": processor}
mm_messages = copy.deepcopy(MM_MESSAGES) mm_messages = copy.deepcopy(MM_MESSAGES)
fake_image_token = processor.fake_image_token.content 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 image_str = image_str * 5
for message in mm_messages: for message in mm_messages:
content = message["content"] content = message["content"]