Fx support for Deberta-v[1-2], Hubert and LXMERT (#17539)

* Support for deberta and deberta-v2

* Support for LXMert

* Support for Hubert

* Fix for pt1.11

* Trigger CI
This commit is contained in:
Michael Benayoun
2022-06-07 18:05:20 +02:00
committed by GitHub
parent 3cab90279f
commit 5c8f601007
10 changed files with 221 additions and 29 deletions

View File

@@ -535,6 +535,7 @@ class LxmertModelTest(ModelTesterMixin, unittest.TestCase):
all_model_classes = (LxmertModel, LxmertForPreTraining, LxmertForQuestionAnswering) if is_torch_available() else ()
fx_compatible = True
test_head_masking = False
test_pruning = False
test_torchscript = False