Unverified Commit d4f201b8 authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

skip test for XGLM (#15445)

parent 0c17e766
......@@ -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)
......
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