[tests] fix the wrong output in...

[tests] fix the wrong output in `ImageToTextPipelineTests.test_conditional_generation_llava` (#29975) bug fix

[tests] fix the wrong output in...
[tests] fix the wrong output in `ImageToTextPipelineTests.test_conditional_generation_llava` (#29975) bug fix
e4f5b57a · Fanli Lin · GitHub · fa2c49b0 · e4f5b57a
Unverified Commit e4f5b57a authored Apr 01, 2024 by Fanli Lin Committed by GitHub Apr 01, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/pipelines/test_pipelines_image_to_text.py tests/pipelines/test_pipelines_image_to_text.py +1 -1

No files found.
--- a/tests/pipelines/test_pipelines_image_to_text.py
+++ b/tests/pipelines/test_pipelines_image_to_text.py
@@ -290,7 +290,7 @@ class ImageToTextPipelineTests(unittest.TestCase):
            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"
                }
            ],
        )