IMvision12 commited on
Commit
fdb36d2
·
verified ·
1 Parent(s): bff60e9

Add Gemma4TextGenerate text-only usage example

Browse files
Files changed (1) hide show
  1. README.md +18 -0
README.md CHANGED
@@ -47,6 +47,24 @@ For model details, license, and usage terms, see Google's
47
 
48
  ## ✨ Quick start
49
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
50
  ```python
51
  import os
52
  os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
 
47
 
48
  ## ✨ Quick start
49
 
50
+ ### Text-only
51
+
52
+ ```python
53
+ import os
54
+ os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
55
+
56
+ from kerasformers.models.gemma4 import Gemma4TextGenerate, Gemma4Tokenizer
57
+
58
+ model = Gemma4TextGenerate.from_weights("kerasformers/gemma-4-31b")
59
+ tokenizer = Gemma4Tokenizer.from_weights("kerasformers/gemma-4-31b")
60
+
61
+ inputs = tokenizer([{"role": "user", "content": "Hello, who are you?"}])
62
+ outputs = model.generate(**inputs, max_new_tokens=64)
63
+ print(tokenizer.decode(outputs[0]))
64
+ ```
65
+
66
+ ### Image + text
67
+
68
  ```python
69
  import os
70
  os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"