Tag tests as slow ⌛ (#21537)

begone slow tests
2023-02-09 14:46:15 +00:00
parent 3a726777ca
commit 0d33381fad
5 changed files with 13 additions and 0 deletions
--- a/tests/models/rag/test_modeling_tf_rag.py
+++ b/tests/models/rag/test_modeling_tf_rag.py
@@ -260,6 +260,7 @@ class TFRagTestMixin:
                num_beams=2,
                num_return_sequences=2,
                decoder_start_token_id=config.generator.eos_token_id,
+                max_new_tokens=5,
            )

            self.assertIsNotNone(outputs)