mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
Enable BLIP for auto VQA (#29499)
* Enable BLIP for auto VQA * Make style * Add VQA to BLIP pipeline tests
This commit is contained in:
parent
d45f47ab7f
commit
979fccc90f
@ -964,6 +964,7 @@ MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING_NAMES = OrderedDict(
|
||||
|
||||
MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPING_NAMES = OrderedDict(
|
||||
[
|
||||
("blip", "BlipForQuestionAnswering"),
|
||||
("blip-2", "Blip2ForConditionalGeneration"),
|
||||
("vilt", "ViltForQuestionAnswering"),
|
||||
]
|
||||
|
@ -432,6 +432,7 @@ class BlipModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
|
||||
{
|
||||
"feature-extraction": BlipModel,
|
||||
"image-to-text": BlipForConditionalGeneration,
|
||||
"visual-question-answering": BlipForQuestionAnswering,
|
||||
}
|
||||
if is_torch_available()
|
||||
else {}
|
||||
|
Loading…
Reference in New Issue
Block a user