Add accelerate support for ViLT (#18683)

2022-09-22 13:14:39 +02:00
parent 9393f966bc
commit 4d0f8c05f5
4 changed files with 12 additions and 7 deletions
--- a/tests/models/vilt/test_modeling_vilt.py
+++ b/tests/models/vilt/test_modeling_vilt.py
@@ -215,7 +215,6 @@ class ViltModelTester:

@require_torch
 class ViltModelTest(ModelTesterMixin, unittest.TestCase):
-
    all_model_classes = (
        (
            ViltModel,
@@ -512,7 +511,6 @@ class ViltModelTest(ModelTesterMixin, unittest.TestCase):

@require_torch
 class ViltForImagesAndTextClassificationModelTest(ViltModelTest, unittest.TestCase):
-
    all_model_classes = (ViltForImagesAndTextClassification,) if is_torch_available() else ()

    def setUp(self):