transformers库使用踩坑记

博主在Win10环境下遇到transformers报错,详细记录了解决过程。问题源于transformers 4.10.0与tokenizers 0.10.3的版本冲突。通过降级tokenizers到0.7.0,然后将transformers降至2.11.0,最终解决了问题。提醒读者并非最新库版本总是最佳选择,稳定版更可靠。
部署运行你感兴趣的模型镜像

今天打算开始试用一下ALBert,安装了transformers之后就一直报错。经过一晚上终于解决了报错。记录一下,给碰到相同问题的小朋友一点参考。

报错代码: 

from transformers import AlbertConfig, AlbertForPreTraining, load_tf_weights_in_albert

报错内容: 

File "D:\python-venv\pytorch-venv\lib\site-packages\transformers\tokenization_utils_base.py", line 74, in <module>
    from tokenizers import AddedToken
  File "D:\python-venv\pytorch-venv\lib\site-packages\tokenizers\__init__.py", line 79, in <module>
    from .tokenizers import (
ImportError: DLL load failed:

这才刚加载就报错,我的系统是Win10,环境为python3.6,采用venv建的虚拟环境。transformers是通过 pip install transformers 直接安装的。

初步怀疑是兼容性问题。尝试单独加载tokenizers库发现报错一样。

from tokenizers import AddedToken

一查tokenizers的版本为0.10.3,难道因为版本太新,于是尝试降级,直至下降到0.7.0才不报错。但是默认安装的transformers的版本为4.10.0,对应tokenizers要求>=0.10.1 <0.11,于是开始降级transformers,直至2.11.0才能使用。

一晚上就这么过去了,类库也不是越新越好,还是要找经典的稳定版本。

您可能感兴趣的与本文相关的镜像

Python3.10

Python3.10

Conda
Python

Python 是一种高级、解释型、通用的编程语言,以其简洁易读的语法而闻名,适用于广泛的应用,包括Web开发、数据分析、人工智能和自动化脚本

评论 4
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值