SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Generate: shorter XLA contrastive search tests (#20354)

Browse Source
This commit is contained in:
Joao Gante
2022-11-22 11:47:12 +00:00
committed by GitHub
parent c3eb01013b
commit f2e7d270ec
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_tf_common.py
View File

@@ -1854,7 +1854,7 @@ class TFModelTesterMixin:
Either the model supports XLA generation and passes the inner test, or it raises an appropriate exception 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 @slow
def test_xla_generate_slow(self): def test_xla_generate_slow(self):