keras加载模型遇到ValueError: Unknown layer: TokenEmbedding

我使用bert进行了文本分类，想使用保存的模型对一些新的数据进行预测

发现会遇到错误ValueError: Unknown layer: TokenEmbedding。

万般搜索不得其解，在网上看到了一个方案https://blog.csdn.net/yscoder/article/details/103214078，尝试之后成功。