【解决】No module named ‘tensorflow.python.keras.engine.keras_tensor‘

项目场景：

在从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。