Hugging Face's logo

google
/
recurrentgemma-2b-flax

Text Generation
RecurrentGemma
JAX
Model card Files Community
2
recurrentgemma-2b-flax / README.md
osanseviero's picture
osanseviero
Add tag to be consistent with transformers
f86aa4d verified
preview code
|
raw
history blame
1.68 kB
metadata
library_name: recurrentgemma
extra_gated_heading: Access RecurrentGemma on Hugging Face
extra_gated_prompt: >-
  To access RecurrentGemma 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
tags:
  - jax
  - recurrent_gemma

RecurrentGemma Model Card

This repository corresponds to the research RecurrentGemma repository in Jax.

Model Page: RecurrentGemma

This model card corresponds to the 2B base version of the RecurrentGemma model for usage with flax. For more information about the model, visit https://huggingface.co/google/recurrentgemma-2b.

Resources and Technical Documentation:

Terms of Use: Terms

Authors: Google

Loading the model

To download the weights and tokenizer, run:

from huggingface_hub import snapshot_download

local_dir = snapshot_download(repo_id="google/recurrentgemma-2b-flax")

# Or to save to a local directory
snapshot_download(repo_id="google/recurrentgemma-2b-flax", local_dir=local_dir)