From e4f5b57a3b43eeb5d292c7e4b3040d9b811b2479 Mon Sep 17 00:00:00 2001 From: Fanli Lin Date: Mon, 1 Apr 2024 19:08:39 +0800 Subject: [PATCH] [tests] fix the wrong output in `ImageToTextPipelineTests.test_conditional_generation_llava` (#29975) bug fix --- tests/pipelines/test_pipelines_image_to_text.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/pipelines/test_pipelines_image_to_text.py b/tests/pipelines/test_pipelines_image_to_text.py index e2d59968ebf4a6..c77353a261f91d 100644 --- a/tests/pipelines/test_pipelines_image_to_text.py +++ b/tests/pipelines/test_pipelines_image_to_text.py @@ -290,7 +290,7 @@ def test_conditional_generation_llava(self): outputs, [ { - "generated_text": " \nUSER: What does the label 15 represent? (1) lava (2) core (3) tunnel (4) ash cloud?\nASSISTANT: Lava" + "generated_text": "\nUSER: What does the label 15 represent? (1) lava (2) core (3) tunnel (4) ash cloud?\nASSISTANT: Lava" } ], )