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

fix some

Browse Source
This commit is contained in:
fzc8578
2025-01-10 20:55:52 +08:00
parent d09032049c
commit 15bbcdf8d3
4 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -281,7 +281,7 @@ class CpmOPlugin(BasePlugin):
mm_inputs = self._get_mm_inputs(images, videos, processor)
pattern = "(<image>./</image>)"
images, image_sizes, _ = mm_inputs["pixel_values"], mm_inputs["image_sizes"], mm_inputs["tgt_sizes"]
images, image_sizes = mm_inputs["pixel_values"], mm_inputs["image_sizes"]
image_index = 0
for index, message in enumerate(messages):
@@ -334,6 +334,7 @@ class CpmOPlugin(BasePlugin):
new_images.append(images[idx : idx + valid_image_nums])
idx += valid_image_nums
images = new_images
image_inputs = image_processor(
images, do_pad=True, max_slice_nums=image_processor.max_slice_nums, return_tensors="pt"
)