根据 here pipeline 提供了一个接口(interface)，可以使用 save_pretrained 方法在本地保存预训练的管道。当我使用它时，我看到一个文件夹是用一堆 json 和 bin 文件创建的，这些文件大概是用于分词器和模型的。
但是文档没有指定加载方法。如何使用本地保存的管道初始化管道？
最佳答案
如果您阅读了save_pretrained 的规范，它只是说明了它
Save[s] the pipeline’s model and tokenizer.
我也给出了一个稍微相关的答案 here关于如何加载自定义模型和分词器。本质上，您可以简单地在 pipeline 中指定特定的模型/路径:
from transformers import pipeline, AutoModel, AutoTokenizer
# Replace with your custom model of choice
model = AutoTokenizer.from_pretrained('/path/to/your/model')
tokenizer = AutoTokenizer.from_pretrained('/path/to/your/tokenizer')
pipe = pipeline(task='summarization', # replace with whatever task you have
model=model,
tokenizer=tokenizer)
关于python - 如何从文件夹加载变压器管道？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/71058732/