Cannot load saved model: TypeError: <class 'keras.src.models.functional.Functional'> could not be deserialized properly. Please ensure that components that are Python object instances (layers, models, etc.) returned by `get_config()` ...

[mw66]

Cannot load saved model:

TypeError: <class 'keras.src.models.functional.Functional'> could not be deserialized properly. Please ensure that components that are Python object instances (layers, models, etc.) returned by get_config() are explicitly deserialized in the model's from_config() method.

[remigenet]

Indeed TKAT model is made using the functionnal API that is not very serialization friendly at all !
This would requires moving the current model from it to creating a subclass of keras.Model instead - not that complicated but I do not have the time to do it for the moment so it is not planned yet, but feel free to do it and I will review the PR.
Maybe I will do it in the future and close this issue.