skip test for XGLM (#15445)

This commit is contained in:
Suraj Patil
2022-01-31 22:53:16 +01:00
committed by GitHub
parent 0c17e766cb
commit d4f201b860

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)