AutoModelForCausalLM.from_pretrained 函数调用本地权重报错

文章目录

1、代码报错的位置(前情提要)

在终端直接运行finetune_lora.sh文件,--model_name_or_path该参数是指定加载的模型权重,可以是huggingface上的model_name,也可以使本地保存模型权重的路径。

使用默认的模型参数配置是(meta-llama/Llama-2-7b-chat-hf),该情况下模型权重会直接去huggingface官网上去下载。

finetune_lora.sh

fintune_clm_lora.py

--model_name_or_path 参数会在这里作为 AutoModelForCausalLM.from_pretrained 函数的参数

2、报错截图

服务器无法直接访问huggingface网站,因此只能采用指定本地模型路径的方法。(需要先申请并下载模型权重)

2.1、huggingfaces上的 meta-llama/Llama-2-7b-chat-hf

这个是需要去meta website申请之后,再在huggingface的meta_llama项目下再提交一次信息才可以查看和下载这些信息

权重文件有【.safetensor】 【.bin】两种,默认是加载.safetensor权重文件

2.2、服务器上模型文件路径

两种权重其实都可以使用,大小都一致,要想只使用.bin权重文件,需要删除.safetensor文件和对应的.safetensor.index.json文件才可以。

3、特别注意事项

若只想使用.bin权重文件,但是.bin.index.json文件有残留,运行就会报错找不到.safetensor权重文件。

可以侧面看出AutoModelForCausalLM.from_pretrained 函数是根据json文件进行读取对应的权重文件(同门李伟发现!)

相关推荐
only-qi3 小时前
大模型微调流程深度解析:从面试题到工程实践
人工智能·机器学习·ai·llm
AINative软件工程3 小时前
LLM 应用的依赖注入工程实践:解耦 Client、Prompt 和 Tool Registry,让 AI 系统真正可测试可替换
后端·llm·前端工程化
ltl13 小时前
PagedAttention 与 Continuous Batching
llm
DigitalOcean19 小时前
在 DigitalOcean 如何提示词缓存:降低 AI 推理成本与延迟
llm·agent
小马92620 小时前
Meta MuseCode:从 Prompt 到持久化 Agent,编码智能体迈入“断点续传“时代
ai·meta·prompt·codingagent·musecode
only-qi21 小时前
美的AI Agent面试题的解析与思考
人工智能·ai·llm·agent·react
武子康21 小时前
VLA 落地先签动作合同:从视觉语言输入到可执行控制指令
人工智能·llm·agent
武子康1 天前
SWE-1.7 的提升究竟来自哪里?接受大量强化学习后训练的 Kimi K2.7 为起点,继续进行大规模 RL
人工智能·llm·agent
用户469368483201 天前
kimi-code 深度掌握系列文章-工具系统:从注册到执行(六)
llm·agent
今日无bug1 天前
用 Prompt 做 NLP 任务开发:几分钟构建一个推理系统
llm·nlp