【解决】No module named ‘tensorflow.python.keras.engine.keras_tensor‘
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。
更多推荐
已为社区贡献1条内容
所有评论(0)