diff --git a/tests/pipelines/test_text_generation.py b/tests/pipelines/test_text_generation.py index 091580dd..2197c4d7 100644 --- a/tests/pipelines/test_text_generation.py +++ b/tests/pipelines/test_text_generation.py @@ -172,7 +172,7 @@ class TextGenerationTest(unittest.TestCase): self.run_pipeline_with_model_id(self.gpt3_large_model_id, self.gpt3_input) - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') def test_hf_model_stream_generate(self): from transformers import AutoTokenizer, GPT2LMHeadModel tokenizer = AutoTokenizer.from_pretrained('gpt2')