SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix (skip) a pipeline test for RwkvModel (#23444)

Browse Source 
fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-05-18 14:54:23 +02:00
committed by GitHub
parent 8cfae44093
commit 5777c3cb3f
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/pipelines/test_pipelines_text_generation.py
View File

@@ -240,7 +240,11 @@ class TextGenerationPipelineTests(unittest.TestCase):
# We don't care about infinite range models. # We don't care about infinite range models.
# They already work. # They already work.
# Skip this test for XGLM, since it uses sinusoidal positional embeddings which are resized on-the-fly. # 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__: EXTRA_MODELS_CAN_HANDLE_LONG_INPUTS = ["RwkvForCausalLM", "XGLMForCausalLM"]
if (
tokenizer.model_max_length < 10000
and text_generator.model.__class__.__name__ not in EXTRA_MODELS_CAN_HANDLE_LONG_INPUTS
):
# 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)