diff --git a/tests/models/opt/test_modeling_tf_opt.py b/tests/models/opt/test_modeling_tf_opt.py index 61d6aad53fc169..4e9972e4aaa195 100644 --- a/tests/models/opt/test_modeling_tf_opt.py +++ b/tests/models/opt/test_modeling_tf_opt.py @@ -315,6 +315,7 @@ def test_logits(self): self.assertTrue(np.allclose(logits, logits_meta, atol=1e-4)) +@require_tf @slow class TFOPTGenerationTest(unittest.TestCase): @property