Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

skip hf download

Browse Source
This commit is contained in:
mulin.lyh
2023-09-11 16:56:50 +08:00
parent 6e38597d4a
commit e297672fa0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/pipelines/test_text_generation.py
View File

@@ -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')