Update README.md

README.md

@@ -1,6 +1,8 @@
 ---
 library_name: keras-nlp
 pipeline_tag: text-generation
+base_model:
+- google/gemma-2-2b-it
 ---
 This is a [`Gemma` model](https://keras.io/api/keras_nlp/models/gemma) uploaded using the KerasNLP library and can be used with JAX, TensorFlow, and PyTorch backends.
 This model is related to a `CausalLM` task.
@@ -25,4 +27,4 @@ Model config:
 * **sliding_window_size:** 4096
 * **use_sliding_window_attention:** True
 
-This model card has been generated automatically and should be completed by the model author. See [Model Cards documentation](https://huggingface.co/docs/hub/model-cards) for more information.
+This model card has been generated automatically and should be completed by the model author. See [Model Cards documentation](https://huggingface.co/docs/hub/model-cards) for more information.