File size: 299 Bytes
357f72e 4a02b16 357f72e |
1 2 3 4 5 6 7 |
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) |