File size: 370 Bytes
c418073 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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",
]
|