No need to to declare the custom_objects.

To save: model.save(filepath)

To load: tf.keras.models.load_model(filepath)