SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source 
bug fix
This commit is contained in:
Fanli Lin
2024-04-01 19:08:39 +08:00
committed by GitHub
parent fa2c49b00b
commit e4f5b57a3b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/pipelines/test_pipelines_image_to_text.py
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"
}
],
)