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 import os
from typing import Optional, Union from typing import Optional, Union
import torch
from transformers import Pipeline as PipelineHF from transformers import Pipeline as PipelineHF
from transformers import PreTrainedModel, TFPreTrainedModel, pipeline from transformers import PreTrainedModel, TFPreTrainedModel, pipeline
from transformers.pipelines import check_task, get_task from transformers.pipelines import check_task, get_task
from modelscope.hub import snapshot_download 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): def _get_hf_device(device):