martin-gorner's picture
Update README.md
d602f54 verified
|
raw
history blame
No virus
2.01 kB
---
library_name: keras-nlp
extra_gated_heading: Access Gemma 1.1 on Hugging Face
extra_gated_prompt: >-
To access Gemma 1.1 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
license_link: https://ai.google.dev/gemma/terms
pipeline_tag: text-generation
---
# Gemma 1.1
This model card corresponds to the latest 7B instruct version of the Gemma 1.1 model in Keras.
Keras models can be used with JAX, PyTorch or TensorFlow as numerical backends.
JAX, with its support for SPMD model paralellism, is recommended for large models.
For more information: [distributed training with Keras and JAX](https://keras.io/guides/distribution/).
You can find other models in the Gemma family here:
| | Base | Instruct |
|----|----------------------------------------------------|----------------------------------------------------------------------|
| 2B | [gemma-2b-keras](https://huggingface.co/google/gemma-2b-keras) | [gemma-1.1-2b-it-keras](https://huggingface.co/google/gemma-1.1-2b-it-keras) |
| 7B | [gemma-7b-keras](https://huggingface.co/google/gemma-7b-keras) | [**gemma-1.1-7b-it-keras**](https://huggingface.co/google/gemma-1.1-7b-it-keras) |
For more information about the model, visit https://huggingface.co/google/gemma-1.1-7b-it.
Model Page
: [Gemma](https://ai.google.dev/gemma/docs/gemma)
Resources and Technical Documentation
: [Technical Report](https://goo.gle/gemma)
: [Responsible Generative AI Toolkit](https://ai.google.dev/responsible)
Terms of Use
: [Terms](https://ai.google.dev/gemma/terms)
Authors
: Google
## Loading the model
```python
import keras_nlp
gemma_lm = keras_nlp.models.GemmaCausalLM.from_preset("hf://google/gemma-1.1-7b-it-keras")
```