本地加载hugging face模型:Bert

找了个hf的镜像站,把config.json和pytorch_model.bin两个文件进行下载下来,模型文件uncased_L-12_H-768_A-12.zip下载下来先。

解压模型文件压缩包,把前面下载的两个文件也放进去,总共6个文件。这个文件夹就是代码

tokenizer = BertTokenizer.from_pretrained(BERT_PATH)

BERT_PATH这里对应的文件路径。

Some weights of BertForSequenceClassification were not initialized from the model checkpoint at bert-base-uncased and are newly initialized: 'classifier.bias', 'classifier.weight'

pip install tensorflow

复制代码
from transformers import BertModel, BertTokenizer

model_name = 'bert-base-uncased'
tokenizer = BertTokenizer.from_pretrained(model_name)
model = BertModel.from_pretrained(model_name, from_tf=True)
相关推荐
九酒5 小时前
AI Agent 开发踩坑记:口播功能非得用 APP 原生实现吗?
前端·人工智能·agent
蝎子莱莱爱打怪5 小时前
DSpark 讲透:DeepSeek 不换模型,硬把 V4 提速 85%,是怎么做到的?
人工智能·面试·程序员
巫山老妖7 小时前
置身AI内
人工智能
IT_陈寒8 小时前
JavaScript项目实战经验分享
前端·人工智能·后端
vanuan10 小时前
两个AI智能体第一次对话-A2A双Agent协作实战
人工智能
kfaino11 小时前
码农的AI翻身(四)你好,我叫 Attention
人工智能·后端
雨落Re14 小时前
如何设计一个高质量Skill
人工智能
Token炼金师14 小时前
大模型权重文件全指南:从格式选择到优化实战
人工智能
阿牛哥_GX14 小时前
CDP 浏览器操控原理:让脚本接管你的浏览器
人工智能
ThreeS14 小时前
手搓MiniVLA全实战教程-一步一步用pytorch解释原理与思路
人工智能·python