ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix some

Browse Source 
Former-commit-id: 1eb7118db3ad6054cfd59d5f16a5d882e40e9057
This commit is contained in:
fzc8578
2025-01-10 20:55:52 +08:00
parent 75b3819e43
commit 40382f1387
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"
)