"benchmark/git@developer.sourcefind.cn:change/sglang.git" did not exist on "9c121f2a45dca269c47812379f851f9ca9478852"
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 ...@@ -152,7 +152,8 @@ class TextGenerationPipelineTests(unittest.TestCase, metaclass=PipelineTestCaseM
return return
# We don't care about infinite range models. # We don't care about infinite range models.
# They already work. # 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 # Handling of large generations
with self.assertRaises((RuntimeError, IndexError, ValueError, AssertionError)): with self.assertRaises((RuntimeError, IndexError, ValueError, AssertionError)):
text_generator("This is a test" * 500, max_new_tokens=20) 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