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

add some

Browse Source 
Former-commit-id: 5ad8ef3ec434f53f6fc494474becb034a3aca0ca
This commit is contained in:
fzc8578 2025-01-11 15:03:20 +08:00
parent a789e0f263
commit 6d6acd0213
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/llamafactory/data/mm_plugin.py
View File

@ -383,6 +383,7 @@ class CpmOPlugin(BasePlugin):
self._validate_input(images, videos) self._validate_input(images, videos)
image_bounds_list = [] image_bounds_list = []
valid_image_nums_ls = [] valid_image_nums_ls = []
for input_ids in batch_ids: for input_ids in batch_ids:
input_ids_ = torch.tensor(input_ids) input_ids_ = torch.tensor(input_ids)
start_cond = (input_ids_ == processor.tokenizer.im_start_id) | ( start_cond = (input_ids_ == processor.tokenizer.im_start_id) | (