Browse Source

update modelhub dependency

master
智丞 3 years ago
parent
commit
e1f058d249
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      tests/pipelines/test_fill_mask.py

+ 1
- 2
tests/pipelines/test_fill_mask.py View File

@@ -3,8 +3,7 @@ import os
import shutil import shutil
import unittest import unittest


from maas_hub.snapshot_download import snapshot_download

from modelscope.hub.snapshot_download import snapshot_download
from modelscope.models import Model from modelscope.models import Model
from modelscope.models.nlp import StructBertForMaskedLM, VecoForMaskedLM from modelscope.models.nlp import StructBertForMaskedLM, VecoForMaskedLM
from modelscope.pipelines import FillMaskPipeline, pipeline from modelscope.pipelines import FillMaskPipeline, pipeline


Loading…
Cancel
Save