tests/pipelines/test_pipelines_visual_question_answering.py · f9296249a3d8fabf9d77d33dbc1105037ead6164 · chenpangpang / transformers

Pipeline VQA: Add support for list of images and questions as pipeline input (#31217) · f9296249

Vu Huy Nguyen authored Jun 06, 2024

* Add list check for image and question

* Handle passing two lists and update docstring

* Add tests

* Add support for dataset

* Add test for dataset as input

* fixup

* fix unprotected import

* fix unprotected import

* fix import again

* fix param type

f9296249

test_pipelines_visual_question_answering.py 9.16 KB

Replace test_pipelines_visual_question_answering.py