在加载权重的过程中出现报错
OSError: Unable to load weights from pytorch checkpoint file for 'checkpoints/llama-7b_hf/pytorch_model-00001-of-00002.bin' at 'checkpoints/llama-7b_hf/pytorch_model-00001-of-00002.bin'. If you tried to load a PyTorch model from a TF 2.0 checkpoint, please set from_tf=True
比然是权重文件损坏了,此时可以先使用代码进行验证
import torch
# 尝试加载单个文件
file_path = "权重文件路径"
try:
state_dict = torch.load(file_path)
print("File loaded successfully.")
except Exception as e:
print(f"Failed to load the file: {e}")
如果能够File loaded successfully,说明文件没问题,如果依然报错如题,则需要重新下载权重文件
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· .NET周刊【3月第1期 2025-03-02】
· [AI/GPT/综述] AI Agent的设计模式综述