docker - Transformers Lib - 使用 Docker 加载预训练模型

Question

我正在尝试使用变压器库（通过拥抱脸）加载预先训练的模型：

from transformers import GPT2Tokenizer, GPT2Model
tokenizer = GPT2Tokenizer.from_pretrained('gpt2-medium')

使用本地机器，它开始下载模型。但是使用 docker 我得到以下信息：

OSError: Model name 'gpt2-medium' was not found in tokenizers model 
name list (gpt2, gpt2-medium, gpt2-large, gpt2-xl, distilgpt2). We 
assumed 'gpt2-medium' was a path, a model identifier, or url to a 
directory containing vocabulary files named ['vocab.json', 
'merges.txt'] but couldn't find such vocabulary files at this path or url.

知道为什么会这样吗？

标签： dockerhuggingface-transformers

Answer 1

这可能意味着您的 docker 实例无法访问互联网