Using a pre-trained Bert model in TF2 as a trainable layer in another TF2 model

Question

Hello Stackoverflow community!

So Google has putout this pre-trained Bert model: https://github.com/google/patents-public-data/blob/master/examples/BERT_For_Patents.i