appledora commited on
Commit
2385ed4
·
verified ·
1 Parent(s): 268f904

Upload __init__.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. __init__.py +31 -0
__init__.py ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers.utils import (
2
+ OptionalDependencyNotAvailable,
3
+ _LazyModule,
4
+ is_torch_available,
5
+ )
6
+
7
+ try:
8
+ if not is_torch_available():
9
+ raise OptionalDependencyNotAvailable()
10
+ except OptionalDependencyNotAvailable:
11
+ pass
12
+ else:
13
+ from .modeling_recast_llama import (
14
+ RECAST1B_llamaModel,
15
+ RECAST1B_LlamaForCausalLM,
16
+ )
17
+ from .configuration_recast_llama import RECAST1B_llama
18
+
19
+ from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
20
+
21
+ # Register your models with Auto classes
22
+ AutoConfig.register("recast1b_llama", RECAST1B_llama)
23
+ AutoModel.register(RECAST1B_llama, RECAST1B_llamaModel)
24
+ AutoModelForCausalLM.register(RECAST1B_llama, RECAST1B_LlamaForCausalLM)
25
+
26
+ _import_structure = {
27
+ "configuration_recastmlp_llama": ["RECAST1B_llama"],
28
+ "modeling_recastmlp_llama": ["RECAST1B_llamaModel", "RECAST1B_LlamaForCausalLM"],
29
+ }
30
+
31
+ __all__ = ["RECAST1B_llamaModel", "RECAST1B_LlamaForCausalLM", "RECAST1B_llama"]