| """Sumeru model class — thin wrapper over Gemma 4 architecture. | |
| Copyright (c) 2026 Sumeru AI. Licensed under Apache 2.0. | |
| Loaded automatically via: | |
| from transformers import AutoModelForImageTextToText, AutoProcessor | |
| model = AutoModelForImageTextToText.from_pretrained( | |
| "debowd/sumeru-rm", trust_remote_code=True | |
| ) | |
| """ | |
| from transformers import Gemma4ForConditionalGeneration | |
| from .configuration_sumeru import SumeruConfig | |
| class SumeruForCausalLM(Gemma4ForConditionalGeneration): | |
| """Sumeru causal language model. | |
| Sumeru is a family of small language models built by Sumeru AI for | |
| Indian education — NCERT curriculum, JEE, and NEET preparation across | |
| Physics, Chemistry, Biology, and Mathematics. | |
| """ | |
| config_class = SumeruConfig | |
| def __init__(self, config): | |
| super().__init__(config) | |
| self.post_init() | |