fix #4742

Former-commit-id: ae9cf84347878fcc462f35db941c14e1df104276
2024-07-09 23:24:24 +08:00
parent 11d961cf3c
commit 834c4e8ad9
2 changed files with 2 additions and 1 deletions
--- a/src/llamafactory/chat/vllm_engine.py
+++ b/src/llamafactory/chat/vllm_engine.py
@@ -135,7 +135,7 @@ class VllmEngine(BaseEngine):
            if is_vllm_version_greater_than_0_5_1():
                multi_modal_data = {"image": pixel_values}
            elif is_vllm_version_greater_than_0_5():
-                multi_modal_data = ImagePixelData(image=pixel_values)   
+                multi_modal_data = ImagePixelData(image=pixel_values)
            else:  # TODO: remove vllm 0.4.3 support
                multi_modal_data = MultiModalData(type=MultiModalData.Type.IMAGE, data=pixel_values)
        else: