python-3.x 变压器管道模型目录

uelo1irk 于 2023-06-25 发布在 Python

关注(0)|答案(2)|浏览(186)

我正在使用Huggingface的Transformers管道功能下载模型和标记器，我的Windows PC下载了它们，但我不知道它们存储在我的PC上的哪里。你能帮帮我吗

from transformers import pipeline

qa_pipeline = pipeline(
    "question-answering",
    model="mrm8488/bert-multi-cased-finetuned-xquadv1",
    tokenizer="mrm8488/bert-multi-cased-finetuned-xquadv1"
)

python-3.x

来源：https://stackoverflow.com/questions/64310515/transformers-pipeline-model-directory

2条答案

按热度按时间

mrzz3bfm1#

您可以使用以下命令检查默认位置：

import transformers #it is important to load the library before checking!
import os
os.environ['TRANSFORMERS_CACHE']

如果您想更改默认位置，请锁定此answer。

赞(0）回复(0）举报 2023-06-25

7lrncoxx2#

模型的默认文件夹在~/.cache/huggingface，这是来自huggingface www.example.com的文档https://huggingface.co/docs/transformers/installation#cache-setup

赞(0）回复(0）举报 2023-06-25

我来回答

python-3.x 变压器管道模型目录

2条答案

相关问题

热门标签

最新问答