Keras loadmodel for custom model with custom layers - Transformer documentation example

Question

I am running the following example:

https://keras.io/examples/nlp/text_classification_with_transformer/

I have created and trained a model as described, and it