HUSTAI / uie_pytorch

PaddleNLP UIE模型的PyTorch版实现
Apache License 2.0
539 stars 93 forks source link

依赖库的版本号 #12

Closed xxllp closed 1 year ago

xxllp commented 1 year ago

请问下transformers等库的版本

heiheiyoyo commented 1 year ago

更新了requirements.txt

xxllp commented 1 year ago

这块uie相关的评论 情感分类的模型 是否也可以转换过来

heiheiyoyo commented 1 year ago

paddle中的uie和uie-m的模型都可以通过convert.py转换,可能需要调整一下convert.py中的代码 python convert.py -i path/to/your/paddle_model -o path/to/output_model

436 if 'uie-m' in args.input_model: 目前判断输入的Paddle模型的类型是通过字符串来判断的,可能需要修改。