Skip to content

Commit 8594417

Browse files
haozha111copybara-github
authored andcommitted
Update Gemma3 model example with 1B decoder.
PiperOrigin-RevId: 748439985
1 parent 8f4d88c commit 8594417

File tree

1 file changed

+1
-1
lines changed
  • ai_edge_torch/generative/examples/gemma3

1 file changed

+1
-1
lines changed

ai_edge_torch/generative/examples/gemma3/gemma3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def forward(
154154
def get_fake_model_config(**kwargs) -> Gemma3MMConfig:
155155
return Gemma3MMConfig(
156156
image_encoder_config=image_encoder.get_fake_image_encoder_config(),
157-
decoder_config=decoder.get_fake_decoder_config_4b(**kwargs),
157+
decoder_config=decoder.get_fake_decoder_config_1b(**kwargs),
158158
image_token_id=127,
159159
image_projection_scale=128**0.5,
160160
image_projection_use_bias=False,

0 commit comments

Comments
 (0)