huggingface-transformers - 拥抱脸的变形金刚在哪里拯救模型？

Question

运行下面的代码会下载一个模型 - 有人知道它下载到哪个文件夹吗？

!pip install -q transformers
from transformers import pipeline
model = pipeline('fill-mask')

标签： huggingface-transformers

Answer 1

2021-03-11 更新：缓存位置现已更改，位于 中~/.cache/huggingface/transformers，@victorx 的回答中也对此进行了详细说明。

---

这篇文章应该对此有所了解（加上我自己的一些调查，因为它已经有点老了）。

如前所述，Linux 系统中的默认位置是~/.cache/torch/transformers/（我目前使用的是 transformers v 2.7，但它不太可能很快改变。）。此目录中的神秘文件夹名称似乎对应于 Amazon S3 哈希。

另请注意，管道任务只是对其他模型的“重新路由”。要知道您当前正在加载哪一个，请参见此处。对于您的特定模型，pipeline(fill-mask)实际使用一个distillroberta-base模型。