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

lint

Browse Source 
Former-commit-id: cee3dc484be4bdf31ffe1d8b9c60604b84ed6a00
This commit is contained in:
hiyouga 2024-09-30 17:00:33 +08:00
parent a3f99f123a
commit b664bcf307
1 changed files with 3 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/data/test_mm_plugin.py
View File

@ -142,10 +142,7 @@ def test_llava_next_plugin():
check_inputs = {"plugin": llava_next_plugin, "tokenizer": tokenizer, "processor": processor}
image_seqlen = 1176
check_inputs["expected_mm_messages"] = [
{
key: value.replace("<image>", "<image>" * image_seqlen)
for key, value in message.items()
}
{key: value.replace("<image>", "<image>" * image_seqlen) for key, value in message.items()}
for message in MM_MESSAGES
]
check_inputs["expected_mm_inputs"] = _get_mm_inputs(processor)
@ -158,10 +155,7 @@ def test_llava_next_video_plugin():
check_inputs = {"plugin": llava_next_video_plugin, "tokenizer": tokenizer, "processor": processor}
image_seqlen = 1176
check_inputs["expected_mm_messages"] = [
{
key: value.replace("<image>", "<image>" * image_seqlen)
for key, value in message.items()
}
{key: value.replace("<image>", "<image>" * image_seqlen) for key, value in message.items()}
for message in MM_MESSAGES
]
check_inputs["expected_mm_inputs"] = _get_mm_inputs(processor)
@ -207,10 +201,7 @@ def test_video_llava_plugin():
check_inputs = {"plugin": video_llava_plugin, "tokenizer": tokenizer, "processor": processor}
image_seqlen = 256
check_inputs["expected_mm_messages"] = [
{
key: value.replace("<image>", "<image>" * image_seqlen)
for key, value in message.items()
}
{key: value.replace("<image>", "<image>" * image_seqlen) for key, value in message.items()}
for message in MM_MESSAGES
]
check_inputs["expected_mm_inputs"] = _get_mm_inputs(processor)