from transformers import AutoConfig, AutoModelForCausalLM from .configuration_veronica import VeronicaConfig from .modeling_veronica import VeronicaForCausalLM # Register with Auto classes AutoConfig.register("veronica", VeronicaConfig) AutoModelForCausalLM.register(VeronicaConfig, VeronicaForCausalLM) __all__ = [ "VeronicaConfig", "VeronicaForCausalLM", ]