rm import

modelscope/utils/hf_util/pipeline_builder.py

@@ -1,10 +1,6 @@
 import os
 from typing import Optional, Union
 
-from transformers import Pipeline as PipelineHF
-from transformers import PreTrainedModel, TFPreTrainedModel, pipeline
-from transformers.pipelines import check_task, get_task
-
 from modelscope.hub import snapshot_download
 from modelscope.utils.hf_util.patcher import _patch_pretrained_class
 
@@ -20,6 +16,7 @@ def _get_hf_device(device):
 
 
 def _get_hf_pipeline_class(task, model):
+    from transformers.pipelines import check_task, get_task
     if not task:
         task = get_task(model)
     normalized_task, targeted_task, task_options = check_task(task)
@@ -34,7 +31,9 @@ def hf_pipeline(
     framework: Optional[str] = None,
     device: Optional[Union[int, str, 'torch.device']] = None,
     **kwargs,
-) -> PipelineHF:
+) -> 'transformers.Pipeline':
+    from transformers import pipeline
+
     if isinstance(model, str):
         if not os.path.exists(model):
             model = snapshot_download(model)