appledora commited on
Commit
b025d9f
·
verified ·
1 Parent(s): 7090542

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
+ RECAST8b_llamaModel,
15
+ RECAST8b_LlamaForCausalLM,
16
+ )
17
+ from .configuration_recast_llama import RECAST8b_llama
18
+
19
+ from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
20
+
21
+ # Register your models with Auto classes
22
+ AutoConfig.register("recast8b_llama", RECAST8b_llama)
23
+ AutoModel.register(RECAST8b_llama, RECAST8b_llamaModel)
24
+ AutoModelForCausalLM.register(RECAST8b_llama, RECAST8b_LlamaForCausalLM)
25
+
26
+ _import_structure = {
27
+ "configuration_recastmlp_llama": ["RECAST8b_llama"],
28
+ "modeling_recastmlp_llama": ["RECAST8b_llamaModel", "RECAST8b_LlamaForCausalLM"],
29
+ }
30
+
31
+ __all__ = ["RECAST8b_llamaModel", "RECAST8b_LlamaForCausalLM", "RECAST8b_llama"]