Former-commit-id: 3a6f19f017539a2be31e348845a8ed8712275dcb
This commit is contained in:
hiyouga 2024-09-02 01:33:22 +08:00
parent 859823eb23
commit f13e974930

View File

@ -208,7 +208,7 @@ class Qwen2vlPlugin(BasePlugin):
) -> List[Dict[str, str]]:
image_processor: "BaseImageProcessor" = getattr(processor, "image_processor")
merge_length: int = getattr(image_processor, "merge_size") ** 2
if len(images) > 0:
if len(images) != 0:
image_grid_thw = _get_mm_inputs(images, processor)["image_grid_thw"]
else:
image_grid_thw = []