diff --git a/modelscope/utils/hf_util/pipeline_builder.py b/modelscope/utils/hf_util/pipeline_builder.py index 5386bead..4a37cd04 100644 --- a/modelscope/utils/hf_util/pipeline_builder.py +++ b/modelscope/utils/hf_util/pipeline_builder.py @@ -1,13 +1,12 @@ import os from typing import Optional, Union -import torch 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, patch_hub +from modelscope.utils.hf_util.patcher import _patch_pretrained_class def _get_hf_device(device):