| """ | |
| SHIVIK-M4 Architecture (Llama-3.2 Compatible) | |
| """ | |
| from transformers import LlamaConfig, LlamaForCausalLM, AutoConfig, AutoModelForCausalLM | |
| class ShivikM4Config(LlamaConfig): | |
| model_type = "shivik_m4" | |
| class ShivikM4ForCausalLM(LlamaForCausalLM): | |
| config_class = ShivikM4Config | |
| # Register for AutoClasses so trust_remote_code works | |
| AutoConfig.register("shivik_m4", ShivikM4Config) | |
| AutoModelForCausalLM.register(ShivikM4Config, ShivikM4ForCausalLM) | |