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",
]