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

bug fix
This commit is contained in:
Fanli Lin 2024-04-01 19:08:39 +08:00 committed by GitHub
parent fa2c49b00b
commit e4f5b57a3b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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"
}
],
)