| 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) |