Transformers管道模型目录

Question

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

from transformers import pipeline

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

- Luan Tran

Answer 1

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

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

如果您想更改默认位置，请查看这个答案。

- cronoik

Answer 2

默认模型文件夹位于~/.cache/huggingface。 这是来自huggingface的文档https://huggingface.co/docs/transformers/installation#cache-setup。

- 杨晨豪

---

虽然这个链接可能回答了问题，但最好在这里包含答案的关键部分，并提供链接作为参考。仅有链接的答案如果链接页面发生变化，就会失效。- 来自评论 - Chenmunka