Explicitly include num_attention_heads and num_key_value_heads in text_config for Gemma3 4B

#44

This makes it easier for downstream users to read directly from config.json without relying on the defaults in Gemma3TextConfig

@lkv @GopiUppari tagging you given your comments on other recent discussions. Is this something we would be able to land?

Ready to merge
This branch is ready to get merged automatically.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment