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

rm import

Browse Source
This commit is contained in:
suluyan
2025-03-10 15:56:52 +08:00
parent 0bddd46cc9
commit 30d8995cd9
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modelscope/utils/hf_util/pipeline_builder.py
View File

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