Cerule-v0.1 / __init__.py
not-lain's picture
update model_type
4a02b16
raw
history blame
299 Bytes
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)