Saving Custom Tensorflow Models

[OmarMAmin]

Hi Team

I'm working with TFRS model from tensorflow, and i've tried to be able to save it in one file, but failed, the issue doesn't seem to be solved yet, so the only way i have to save the model is to save the weights and then recreate the model and load the saved weights

# Save the weights
model.save_weights('./checkpoints/my_checkpoint')

# Create a new model instance
model = create_model()

# Restore the weights
model.load_weights('./checkpoints/my_checkpoint')

Is there a way to support saving the models in a two step manner like that? as i guess bentoml.tensorflow.save_model is calling the save method of the model object