Views
No views yet
-it / 1.1 variants are instruction-tuned for chat.google/gemma-7b-it for zeromodels. One implementation runs unmodified on TensorFlow / Torch / JAX.GemmaTokenizer.1import os
2os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
3
4from zeromodels.models.gemma import GemmaTextGenerate, GemmaTokenizer
5
6model = GemmaTextGenerate.from_weights("zeromodels/gemma-7b-it")
7tokenizer = GemmaTokenizer.from_weights("zeromodels/gemma-7b-it")
8
9inputs = tokenizer([
10 {"role": "user", "content": "Explain rotary embeddings in one sentence."}
11])
12outputs = model.generate(**inputs, max_new_tokens=64)
13print(tokenizer.decode(outputs[0]))from_weights("zeromodels/<variant>"):| Variant | Hub | Type |
|---|---|---|
gemma-2b | zeromodels/gemma-2b | base |
gemma-2b-it | zeromodels/gemma-2b-it | instruct |
gemma-1.1-2b-it | zeromodels/gemma-1.1-2b-it | instruct (1.1) |
gemma-7b | zeromodels/gemma-7b | base |
gemma-7b-it | zeromodels/gemma-7b-it | instruct |
gemma-1.1-7b-it | zeromodels/gemma-1.1-7b-it | instruct (1.1) |
KERAS_BACKEND before importing Keras / zeromodels.GemmaTokenizer.from_weights(...) so the chat template matches.load_dtype="bfloat16" or quantization="int8".hf: prefix, e.g. GemmaTextGenerate.from_weights("hf:google/gemma-7b-it").