lsy76

导航

< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5
统计
 

在加载权重的过程中出现报错
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,说明文件没问题,如果依然报错如题,则需要重新下载权重文件

posted on   tujubao  阅读(380)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· .NET周刊【3月第1期 2025-03-02】
· [AI/GPT/综述] AI Agent的设计模式综述
 
点击右上角即可分享
微信分享提示