lbs1163 commited on
Commit
d60eb70
·
verified ·
1 Parent(s): 82ab4a2

Create proxy_bert.py

Browse files
Files changed (1) hide show
  1. proxy_bert.py +14 -0
proxy_bert.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ try:
2
+ from mblt_model_zoo.hf_transformers.models.bert.configuration_bert import (
3
+ MobilintBertConfig,
4
+ )
5
+ from mblt_model_zoo.hf_transformers.models.bert.modeling_bert import (
6
+ MobilintBertForMaskedLM,
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__ = ["MobilintBertConfig", "MobilintBertForMaskedLM"]