ValueError: tf.function-decorated function tried to create variables on non-first call

Question

I have a model written with Model Subclassing API in tensorflow 2. It contains a custom layer. The issue is, in the custom layer, I need to sen