423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-03 04:02:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix ci

Browse Source 
Former-commit-id: b52c38350d6092e8158484ca01da19ed016236fc
This commit is contained in:
hiyouga 2024-11-23 19:13:32 +00:00
parent 8dff6f630c
commit 4bba121f43
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/data/test_mm_plugin.py
View File

@ -61,7 +61,7 @@ INPUT_IDS = [0, 1, 2, 3, 4]
LABELS = [0, 1, 2, 3, 4] LABELS = [0, 1, 2, 3, 4]
SEQLENS = [1024] BATCH_IDS = [[1] * 1024]
def _get_mm_inputs(processor: "ProcessorMixin") -> Dict[str, "torch.Tensor"]: def _get_mm_inputs(processor: "ProcessorMixin") -> Dict[str, "torch.Tensor"]:
@ -105,7 +105,7 @@ def _check_plugin(
expected_labels, expected_labels,
) )
_is_close( _is_close(
plugin.get_mm_inputs(IMAGES, NO_VIDEOS, IMGLENS, NO_VIDLENS, SEQLENS, processor), plugin.get_mm_inputs(IMAGES, NO_VIDEOS, IMGLENS, NO_VIDLENS, BATCH_IDS, processor),
expected_mm_inputs, expected_mm_inputs,
) )
# test text_messages # test text_messages
@ -115,7 +115,7 @@ def _check_plugin(
LABELS, LABELS,
) )
_is_close( _is_close(
plugin.get_mm_inputs(NO_IMAGES, NO_VIDEOS, NO_IMGLENS, NO_VIDLENS, SEQLENS, processor), plugin.get_mm_inputs(NO_IMAGES, NO_VIDEOS, NO_IMGLENS, NO_VIDLENS, BATCH_IDS, processor),
expected_no_mm_inputs, expected_no_mm_inputs,
) )