__init__.py

from .configuration_cerule_gemma import CeruleGemmaConfig
from .modeling_cerule_gemma import CeruleGemmaForCausalLM

from transformers import AutoConfig, AutoModelForCausalLM

AutoConfig.register("cerule-gemma", CeruleGemmaConfig)
AutoModelForCausalLM.register(CeruleGemmaConfig, CeruleGemmaForCausalLM)