using the decorator:

@keras.saving.register_keras_serializable(package="name")

and returning init params in get_config, I had to rerun the notebook and retrain the model.

Now I can successfully load the model using:

model2 = keras.models.load_model("model_1_epoch.keras", custom_objects=custom_objects, compile=False)