lbs1163 commited on
Commit
73bb32c
·
verified ·
1 Parent(s): a496ecf

Create proxy_llama.py

Browse files
Files changed (1) hide show
  1. proxy_llama.py +14 -0
proxy_llama.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ try:
2
+ from mblt_model_zoo.hf_transformers.models.llama.configuration_llama import (
3
+ MobilintLlamaConfig,
4
+ )
5
+ from mblt_model_zoo.hf_transformers.models.llama.modeling_llama import (
6
+ MobilintLlamaForCausalLM,
7
+ )
8
+ except ImportError:
9
+ raise ImportError(
10
+ "This model requires 'mblt_model_zoo' to be installed. "
11
+ "Please run: pip install mblt_model_zoo[transformers]"
12
+ )
13
+
14
+ __all__ = ["MobilintLlamaConfig", "MobilintLlamaForCausalLM"]