Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "accccdd0087263a1e494e9c9ec30a43043ff3905"
Unverified Commit f2e7d270 authored by Joao Gante's avatar Joao Gante Committed by GitHub
Browse files

Generate: shorter XLA contrastive search tests (#20354)

parent c3eb0101
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_modeling_tf_common.py
View file @ f2e7d270
...@@ -1854,7 +1854,7 @@ class TFModelTesterMixin: ...@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment