Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 12:39:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

pre-commit lint

Browse Source
This commit is contained in:
雨泓
2022-06-23 13:12:57 +08:00
parent 7d9b163a92
commit 56cb04cefe
26 changed files with 103 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/pipelines/test_word_segmentation.py
View File

@@ -19,8 +19,7 @@ class WordSegmentationTest(unittest.TestCase):
def test_run_by_direct_model_download(self):
cache_path = snapshot_download(self.model_id)
tokenizer = TokenClassifcationPreprocessor(cache_path)
model = SbertForTokenClassification(
cache_path, tokenizer=tokenizer)
model = SbertForTokenClassification(cache_path, tokenizer=tokenizer)
pipeline1 = WordSegmentationPipeline(model, preprocessor=tokenizer)
pipeline2 = pipeline(
Tasks.word_segmentation, model=model, preprocessor=tokenizer)