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

rm comment

Browse Source 
Former-commit-id: c1d17b0f013f40b48322cb13979fe3726dc124a6
This commit is contained in:
KUANGDD 2024-10-23 15:50:59 +08:00
parent b9c6fcfe98
commit 9d1f079ca5
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -510,7 +510,6 @@ class PixtralPlugin(BasePlugin):
) -> Dict[str, Union[List[int], "torch.Tensor"]]: ) -> Dict[str, Union[List[int], "torch.Tensor"]]:
self._validate_input(images, videos) self._validate_input(images, videos)
mm_inputs = self._get_mm_inputs(images, videos, processor) mm_inputs = self._get_mm_inputs(images, videos, processor)
# hack for hf engine
if mm_inputs.get("pixel_values"): if mm_inputs.get("pixel_values"):
mm_inputs["pixel_values"] = mm_inputs["pixel_values"][0] mm_inputs["pixel_values"] = mm_inputs["pixel_values"][0]