【transformers】pipeline_transformers pipeline 多进程卡住

☆ 问题描述

transformers库中提供了多种快捷的方法，pipeline就是其中的一个高级方法，中文意思是“管道”。
pipeline提供了快捷的模型加载和使用

★ 解决方案

一些常用的代码

# 查看支持的任务类型
from transformers.pipelines import SUPPORTED_TASKS

# 创建pipeline
from transformers imoport pipeline
pipe = pipeline("text-classification")  # 括号里加任务名

# 指定任务和模型
pipe = pipeline("text-classification", model="uer/roberta-base-finetuned-dianping-chinese")

1
2
3
4
5
6
7
8
9
10

✅ 总结