from .configuration_cerule_gemma import CeruleGemmaConfig
from .modeling_cerule_gemma import CeruleGemmaForCausalLM

from transformers import AutoConfig, AutoModelForCausalLM

AutoConfig.register("phi-msft", CeruleGemmaConfig)
AutoModelForCausalLM.register(CeruleGemmaConfig, CeruleGemmaForCausalLM)