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

remove bs condition

Browse Source 
Former-commit-id: bf3520178ab66058c62a9cf31b42f36a9d88ce20
This commit is contained in:
Kingsley 2024-10-14 16:55:59 +08:00
parent 313a5e1494
commit 9d929897ce
1 changed files with 2 additions and 7 deletions
Show Stats 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"): if mm_inputs.get("image_sizes"):
mm_inputs.pop("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 return mm_inputs
class Qwen2vlPlugin(BasePlugin): class Qwen2vlPlugin(BasePlugin):