项目场景:

在从hugging——Faca下载好bert-base-chinese,进行简单使用时

from transformers import BertTokenizer, TFBertModel
tokenizer = BertTokenizer.from_pretrained('bert-base-chinese')
model = TFBertModel.from_pretrained("bert-base-chinese")
text = "Replace me by any text you'd like."
encoded_input = tokenizer(text, return_tensors='tf')
output = model(encoded_input)

问题描述

RuntimeError: Failed to import transformers.models.bert.modeling_tf_bert because of the following error (look up to see its traceback):
No module named 'tensorflow.python.keras.engine.keras_tensor'


原因分析:

因为我们只导入了transformers库,所以考虑是transformers和tensorflow或者keras不兼容的问题。以下是我使用的版本。

Keras                    2.3.1
tensorflow-gpu           2.2.0
tensorflow-gpu-estimator 2.2.0

transformers             4.19.2
tokenizers               0.12.1

解决方案:

更换比较稳定的transformers==3.4.0 版本,再安装之前记得先卸载之前的transformers。

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐