Tensorflow strategy error: Mixing different tf.distribute.Strategy objects with overrides train_step keras model

Question

I am having difficulties to implement a distributed strategy on my TensorFlow v2 model and I am looking for some help.
The main difference with the documentation is th