本地加载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)
相关推荐
DS随心转插件几秒前
DeepSeek 代码手机端导出与 AI 辅助方案实测
android·人工智能·chatgpt·智能手机·deepseek·ai导出鸭
He少年几秒前
【实践AI 编程规则引擎深度解读:从零理解工程化设计】
人工智能
hero827482741 分钟前
公司武装AI
人工智能
阿文的代码库5 分钟前
机器学习任务二分类的应用案例
人工智能·机器学习·分类
hixiong1237 分钟前
C# LLamaSharp部署大语言模型实例
人工智能·语言模型·自然语言处理
morning_judger12 分钟前
Agent开发系列(八)-组织知识库建设
人工智能
HERR_QQ13 分钟前
端到端课程自用 8 规划 端到端与VLA 世界模型 RL的关系
人工智能·深度学习·自动驾驶·transformer
小江的记录本13 分钟前
【Spring全家桶】Spring AI核心原理、大模型集成、Prompt工程、RAG实现、AI Agent开发(附《思维导图》+《面试高频考点清单》)
java·人工智能·spring boot·后端·spring·面试·prompt
jiayong2318 分钟前
AI工作流实现原理深度解析
人工智能·comfyui·工作流·coze
用户51914958484518 分钟前
Nortek Linear eMerge E3 预认证远程代码执行漏洞利用工具
人工智能·aigc