MhaWay commited on
Commit
c418073
·
verified ·
1 Parent(s): 2c26b1e

Create __init__.py

Browse files
Files changed (1) hide show
  1. src/veronica/__init__.py +13 -0
src/veronica/__init__.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import AutoConfig, AutoModelForCausalLM
2
+
3
+ from .configuration_veronica import VeronicaConfig
4
+ from .modeling_veronica import VeronicaForCausalLM
5
+
6
+ # Register with Auto classes
7
+ AutoConfig.register("veronica", VeronicaConfig)
8
+ AutoModelForCausalLM.register(VeronicaConfig, VeronicaForCausalLM)
9
+
10
+ __all__ = [
11
+ "VeronicaConfig",
12
+ "VeronicaForCausalLM",
13
+ ]