tests/pipelines/test_pipelines_visual_question_answering.py · 09dc99517f5f38ee210cf1145a7b17fc99b37dac · chenpangpang / transformers

Add Blip2 model in VQA pipeline (#25532) · 09dc9951

Juan Pizarro authored Aug 30, 2023

* Add Blip2 model in VQA pipeline

* use require_torch_gpu for test_large_model_pt_blip2

* use can_generate in vqa pipeline

* test Blip2ForConditionalGeneration using float16

* remove custom can_generate from Blip2ForConditionalGeneration

09dc9951

test_pipelines_visual_question_answering.py 6.45 KB

Replace test_pipelines_visual_question_answering.py