metadata

library_name: keras-nlp
extra_gated_heading: Access Gemma on Hugging Face
extra_gated_prompt: >-
  To access Gemma on Hugging Face, you’re required to review and agree to
  Google’s usage license. To do this, please ensure you’re logged-in to Hugging
  Face and click below. Requests are processed immediately.
extra_gated_button_content: Acknowledge license
license: gemma
pipeline_tag: text-generation

Gemma Model Card

Model Page: Gemma

This model card corresponds to the latest 2B version of the Gemma 1.1 model in Keras. You can find other models in the Gemma family here:

	Base	Instruct
2B	gemma-2b-keras	gemma-1.1-2b-it-keras
7B	gemma-7b-keras	gemma-1.1-7b-it-keras

For more information about the model, visit https://huggingface.co/google/gemma-2b.

Resources and Technical Documentation:

Responsible Generative AI Toolkit
Gemma on Kaggle
Gemma on Vertex Model Garden

Terms of Use: Terms

Authors: Google

Loading the model

import keras_nlp

gemma_lm = keras_nlp.models.GemmaCausalLM.from_preset("hf://google/gemma-2b-keras")