Add skip logic for attentions test - Levit (#17633)

This commit is contained in:
amyeroberts 2022-06-10 12:46:30 +02:00 committed by GitHub
parent cdaed367b0
commit fd1e67033e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -205,6 +205,10 @@ class LevitModelTest(ModelTesterMixin, unittest.TestCase):
def test_model_common_attributes(self):
pass
@unittest.skip(reason="Levit does not output attentions")
def test_attention_outputs(self):
pass
def test_forward_signature(self):
config, _ = self.model_tester.prepare_config_and_inputs_for_common()