Unverified Commit 979fccc9 authored by regisss's avatar regisss Committed by GitHub
Browse files

Enable BLIP for auto VQA (#29499)

* Enable BLIP for auto VQA

* Make style

* Add VQA to BLIP pipeline tests
parent d45f47ab
...@@ -964,6 +964,7 @@ MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING_NAMES = OrderedDict( ...@@ -964,6 +964,7 @@ MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING_NAMES = OrderedDict(
MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPING_NAMES = OrderedDict( MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPING_NAMES = OrderedDict(
[ [
("blip", "BlipForQuestionAnswering"),
("blip-2", "Blip2ForConditionalGeneration"), ("blip-2", "Blip2ForConditionalGeneration"),
("vilt", "ViltForQuestionAnswering"), ("vilt", "ViltForQuestionAnswering"),
] ]
......
...@@ -432,6 +432,7 @@ class BlipModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase): ...@@ -432,6 +432,7 @@ class BlipModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
{ {
"feature-extraction": BlipModel, "feature-extraction": BlipModel,
"image-to-text": BlipForConditionalGeneration, "image-to-text": BlipForConditionalGeneration,
"visual-question-answering": BlipForQuestionAnswering,
} }
if is_torch_available() if is_torch_available()
else {} else {}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment