Enable BLIP for auto VQA (#29499)

* Enable BLIP for auto VQA * Make style * Add VQA to BLIP pipeline tests
2024-03-07 10:28:01 +01:00
parent d45f47ab7f
commit 979fccc90f
2 changed files with 2 additions and 0 deletions
--- a/tests/models/blip/test_modeling_blip.py
+++ b/tests/models/blip/test_modeling_blip.py
@@ -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 {}