import build_transformer_model时报错

[whats0mai0name]

大佬您好，我在尝试import build_transformer_model的时候报错了，报错提示‘ModuleNotFoundError: No module named 'keras.api'’，尝试使用pip install keras==2.9.0但是依然报错，请问这种情况应该怎么解决呢？系统的基本信息如下：

基本信息

• 你使用的操作系统: 使用的是colab
• 你使用的Python版本: 3.7
• 你使用的Tensorflow版本: 2.9.0
• 你使用的Keras版本: 2.3.1
• 你使用的bert4keras版本: 0.11.4

核心代码

from bert4keras.models import build_transformer_model

### 输出信息
ModuleNotFoundError: No module named 'keras.api'

[bojone]

keras不超过2.3.1，tf不超过2.6，最好用tf 1.15