首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >如何在本地下载BERT模型,不使用包?

如何在本地下载BERT模型,不使用包?
EN

Stack Overflow用户
提问于 2021-04-14 01:58:34
回答 1查看 218关注 0票数 0

公司防火墙似乎阻止我只使用

代码语言:javascript
复制
model = AutoModel.from_pretrained("sentence-transformers/bert-base-nli-stsb-mean-tokens")

因此,我需要在本地下载此模型,然后将其读取到Python中。找不到直接的AWS链接,似乎通常是这种形式:但不起作用

代码语言:javascript
复制
https://s3.amazonaws.com/models.huggingface.co/bert/bert-base-nli-stsb-mean-tokens-pytorch_model.bin

尝试了这些类似的问题/解决方案here,但不起作用,因为我不能运行第一行下载从预先训练的Python语言,我需要一个外部解决方案

EN

回答 1

Stack Overflow用户

发布于 2021-08-08 13:58:03

您可以尝试这样做,例如:

代码语言:javascript
复制
from transformers import AutoTokenizer, TFAutoModel

tokenizer = AutoTokenizer.from_pretrained('bert-base-multilingual-uncased')
bert=TFAutoModel.from_pretrained('bert-base-multilingual-uncased')

tokenizer.save_pretrained("./models/tokenizer/")
bert.save_pretrained("./models/tokenizer/")
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/67080085

复制
相关文章

相似问题

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