From f9daf1187f1bed537e8f3085eede888d277068e6 Mon Sep 17 00:00:00 2001
From: Amy Roberts <22614925+amyeroberts@users.noreply.github.com>
Date: Wed, 31 Jan 2024 20:02:18 +0000
Subject: [PATCH] Add back FE pipeline for Vilt

---
 tests/models/vilt/test_modeling_vilt.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/models/vilt/test_modeling_vilt.py b/tests/models/vilt/test_modeling_vilt.py
index fd93d9ab5821ac..ac11993250a9dc 100644
--- a/tests/models/vilt/test_modeling_vilt.py
+++ b/tests/models/vilt/test_modeling_vilt.py
@@ -227,7 +227,7 @@ class ViltModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
         if is_torch_available()
         else ()
     )
-    pipeline_model_mapping = {"visual-question-answering": ViltForQuestionAnswering} if is_torch_available() else {}
+    pipeline_model_mapping = {"image-feature-extraction": ViltModel, "visual-question-answering": ViltForQuestionAnswering} if is_torch_available() else {}
     test_pruning = False
     test_headmasking = False
     test_torchscript = False