From f2e7d270ec795be09e6187dd2459edb43bd861c1 Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Tue, 22 Nov 2022 11:47:12 +0000 Subject: [PATCH] Generate: shorter XLA contrastive search tests (#20354) --- tests/test_modeling_tf_common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_tf_common.py b/tests/test_modeling_tf_common.py index c888ec7c8c..15d4357f0b 100644 --- a/tests/test_modeling_tf_common.py +++ b/tests/test_modeling_tf_common.py @@ -1854,7 +1854,7 @@ class TFModelTesterMixin: Either the model supports XLA generation and passes the inner test, or it raises an appropriate exception """ - self._test_xla_generate(num_beams=1, num_return_sequences=1, max_new_tokens=64, penalty_alpha=0.5, top_k=4) + self._test_xla_generate(num_beams=1, num_return_sequences=1, max_new_tokens=16, penalty_alpha=0.5, top_k=4) @slow def test_xla_generate_slow(self):