riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

skip test for XGLM (#15445)

Browse Source
This commit is contained in:
Suraj Patil 2022-01-31 22:53:16 +01:00 committed by GitHub
parent 0c17e766cb
commit d4f201b860
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_pipelines_text_generation.py
View File

@ -152,7 +152,8 @@ class TextGenerationPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseM
return
# We don't care about infinite range models.
# They already work.
if tokenizer.model_max_length < 10000:
# Skip this test for XGLM, since it uses sinusoidal positional embeddings which are resized on-the-fly.
if tokenizer.model_max_length < 10000 and "XGLM" not in tokenizer.__class__.__name__:
# Handling of large generations
with self.assertRaises((RuntimeError, IndexError, ValueError, AssertionError)):
text_generator("This is a test" * 500, max_new_tokens=20)