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

add some

Browse Source 
Former-commit-id: 01e9cfd406dc21f387b4f2baa1d61195a841ccb5
This commit is contained in:
fzc8578 2025-01-11 15:03:20 +08:00
parent 8c79fe6a5a
commit bdded9d41a
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) | (