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

remove bs condition

Browse Source
This commit is contained in:
Kingsley
2024-10-14 16:55:59 +08:00
parent 95330893c5
commit 962b9730a7
1 changed files with 2 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -516,11 +516,6 @@ class PixtralPlugin(BasePlugin):
if mm_inputs.get("image_sizes"):
mm_inputs.pop("image_sizes")
if isinstance(mm_inputs.get("pixel_values"), list) and len(mm_inputs.get("pixel_values")[0]) >= 2:
raise ValueError("Now it only supports batchsize=1 on per gpu due to `List[tensor]` can not pack into BachEncoding")
mm_inputs["pixel_values"] = mm_inputs.get("pixel_values")[0][0].unsqueeze(0)
return mm_inputs
class Qwen2vlPlugin(BasePlugin):