首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >使用Docker的Transformers - Load预训练模型

使用Docker的Transformers - Load预训练模型
EN

Stack Overflow用户
提问于 2020-09-15 21:03:35
回答 1查看 201关注 0票数 0

我正在尝试使用transformers lib加载一个预先训练好的模型(通过拥抱脸):

代码语言:javascript
运行
复制
from transformers import GPT2Tokenizer, GPT2Model
tokenizer = GPT2Tokenizer.from_pretrained('gpt2-medium')

使用本地机器,它开始下载模型。但使用docker时,我得到的结果如下:

代码语言:javascript
运行
复制
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.

你知道为什么会这样吗?

EN

回答 1

Stack Overflow用户

发布于 2020-10-14 22:40:15

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

票数 -1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/63902455

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档