tiny fix

2024-09-02 01:33:22 +08:00 · 2024-09-02 01:33:22 +08:00 · 3a6f19f017
parent ce8c5a2647
commit 3a6f19f017
1 changed files with 1 additions and 1 deletions
--- a/src/llamafactory/data/mm_plugin.py
+++ b/src/llamafactory/data/mm_plugin.py
@ -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 = []