From bdded9d41a1c7900d5b84f2c9e33d32000419400 Mon Sep 17 00:00:00 2001 From: fzc8578 <1428195643@qq.com> Date: Sat, 11 Jan 2025 15:03:20 +0800 Subject: [PATCH] add some Former-commit-id: 01e9cfd406dc21f387b4f2baa1d61195a841ccb5 --- src/llamafactory/data/mm_plugin.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/llamafactory/data/mm_plugin.py b/src/llamafactory/data/mm_plugin.py index 55db4626..3de51904 100644 --- a/src/llamafactory/data/mm_plugin.py +++ b/src/llamafactory/data/mm_plugin.py @@ -383,6 +383,7 @@ class CpmOPlugin(BasePlugin): self._validate_input(images, videos) image_bounds_list = [] valid_image_nums_ls = [] + for input_ids in batch_ids: input_ids_ = torch.tensor(input_ids) start_cond = (input_ids_ == processor.tokenizer.im_start_id) | (