| """Sumeru model class — thin wrapper over Gemma 4 architecture. | |
| Copyright (c) 2026 Sumeru AI. Licensed under Apache 2.0. | |
| Usage: | |
| from transformers import AutoModelForCausalLM, AutoProcessor | |
| processor = AutoProcessor.from_pretrained("debowd/sumeru-i", trust_remote_code=True) | |
| model = AutoModelForCausalLM.from_pretrained("debowd/sumeru-i", 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 — math, science, NCERT curriculum, JEE and NEET prep. | |
| """ | |
| config_class = SumeruConfig | |
| def __init__(self, config): | |
| super().__init__(config) | |
| self.post_init() | |