unilm 在使用TrOCR识别文本时出现错误

wooyq4lh  于 2个月前  发布在  其他
关注(0)|答案(2)|浏览(40)

描述

我已检出提交的**65f15af**,并使用模型(trocr-large-stage1.pt)进行文本识别。遇到了以下错误:
ImportError: 无法从'fairseq.modules.transformer_layer'导入名称'TransformerDecoderLayerBase' (unilm/.venv/lib/python3.8/site-packages/fairseq/modules/transformer_layer.py)

wgxvkvu9

wgxvkvu92#

fairseq 的安装版本使用了需求,它很旧并且没有安装 TransformerDecoderLayerBase 。要解决这个问题,您必须卸载当前版本,然后按照 fairseq 官方GitHub存储库上的安装说明进行操作。
您需要执行的操作:

pip uninstall fairseq
git clone https://github.com/pytorch/fairseq
cd fairseq
pip install --editable ./

相关问题