[tests] fix the wrong output in `ImageToTextPipelineTests.test_condit…

…ional_generation_llava` (#29975) bug fix
huggingface · Apr 22, 2024 · ecaa3bc · ecaa3bc
1 parent 8ef7a98
commit ecaa3bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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": "<image> \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"
                 }
             ],
         )