Skip to content

Commit 82df7d3

Browse files
authored
Point to Gemma 4 model in Gemma4ForCausalLM docstring example (#47255)
Point to gemma 4 model in Gemma4ForCausalLM docstring example
1 parent 11ed2ff commit 82df7d3

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

src/transformers/models/gemma4/modeling_gemma4.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1866,8 +1866,8 @@ def forward(
18661866
```python
18671867
>>> from transformers import AutoTokenizer, Gemma4ForCausalLM
18681868
1869-
>>> model = Gemma4ForCausalLM.from_pretrained("google/gemma-2-9b")
1870-
>>> tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-9b")
1869+
>>> model = Gemma4ForCausalLM.from_pretrained("google/gemma-4-E2B-it")
1870+
>>> tokenizer = AutoTokenizer.from_pretrained("google/gemma-4-E2B-it")
18711871
18721872
>>> prompt = "What is your favorite condiment?"
18731873
>>> inputs = tokenizer(prompt, return_tensors="pt")

src/transformers/models/gemma4/modular_gemma4.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1597,8 +1597,8 @@ def forward(
15971597
```python
15981598
>>> from transformers import AutoTokenizer, Gemma4ForCausalLM
15991599
1600-
>>> model = Gemma4ForCausalLM.from_pretrained("google/gemma-2-9b")
1601-
>>> tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-9b")
1600+
>>> model = Gemma4ForCausalLM.from_pretrained("google/gemma-4-E2B-it")
1601+
>>> tokenizer = AutoTokenizer.from_pretrained("google/gemma-4-E2B-it")
16021602
16031603
>>> prompt = "What is your favorite condiment?"
16041604
>>> inputs = tokenizer(prompt, return_tensors="pt")

0 commit comments

Comments
 (0)