当前位置:首页 » 《关注互联网》 » 正文

加载模型时出现 OSError: Unable to load weights from pytorch checkpoint file 报错的解决

24 人参与  2024年04月02日 09:16  分类 : 《关注互联网》  评论

点击全文阅读


加载模型时出现 OSError: Unable to load weights from pytorch checkpoint file 报错的解决

报错信息原因查明网传解决措施好消息我的解决措施

报错信息

查了下,在网上还是个比较常见的报错
一般为加载某模型时突然报错
在这里插入图片描述

原因查明

一般为下载某个 XXX_model.bin 的时候下载报错了
而下载源可以看到为 HuggingFace

网传解决措施

首先得看看你是下载哪个模型报错了。像我这里有一个中间模型,所以不知道是哪个低层模型下载报错了。
打开 ~/.cache/huggingface/hub,这里 ~ 表示本机用户名
若有多个模型,你可以看一下修改日期是今天的为哪个
在这里插入图片描述
这里发现,是模型 potsawee--t5-large-generation-squad-QuestionAnswer
HuggingFace 上搜索相应模型,自行下载,比如我这里即为 网址
在这里插入图片描述
该文件下载到刚刚那个缓存文件夹中对应的放那些files的文件夹。做法二:删除该缓存文件夹中所有东西,再运行一遍代码。

好消息

虽然貌似没有报错,但是加载 selfcheck_mqag 模型的时候内存完全吃完了,卡到死机也没跑完初始化,可能是没问题了只是太吃内存了
对于一般的小模型该报错可能可以直接解决了。

我的解决措施

既然本机实在是跑不了,大不了不跑本机,去 Colab 上跑
见我的博客,关于Colab的简单应用

首先代码放进去,然后先跑个 install 包命令
在这里插入图片描述

然后再运行代码
在这里插入图片描述

发现对于所有的文件,包括 bin 文件都成功安装。
初始化成功,且不吃本机资源, n i c e nice nice
在这里插入图片描述

yy:MQAG 也太 tm 吃资源了


点击全文阅读


本文链接:http://zhangshiyu.com/post/89292.html

<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

关于我们 | 我要投稿 | 免责申明

Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1