Save and Load a Keras NLP Tuned model

[msmmpts]

Hi,

I was following this tutorial

Link : https://colab.research.google.com/github/google/generative-ai-docs/blob/main/site/en/gemma/docs/lora_tuning.ipynb

Could please tell me as to how to save a fine tuned model to disk and load it back from Keras NLP?

[RYangData]

gemma_lm.save_weights("/dir/model.weights.h5") gemma_lm.preprocessor.tokenizer.save_assets("dir")