Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
d4f201b8
"benchmark/git@developer.sourcefind.cn:change/sglang.git" did not exist on "9c121f2a45dca269c47812379f851f9ca9478852"
Unverified
Commit
d4f201b8
authored
Jan 31, 2022
by
Suraj Patil
Committed by
GitHub
Jan 31, 2022
Browse files
skip test for XGLM (#15445)
parent
0c17e766
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
tests/test_pipelines_text_generation.py
tests/test_pipelines_text_generation.py
+2
-1
No files found.
tests/test_pipelines_text_generation.py
View file @
d4f201b8
...
@@ -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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment