lorenzozan commited on
Commit
ec028fe
·
verified ·
1 Parent(s): 46a120f

Update configuration_me2bert.py

Browse files
Files changed (1) hide show
  1. configuration_me2bert.py +4 -0
configuration_me2bert.py CHANGED
@@ -3,6 +3,10 @@ from transformers import PretrainedConfig
3
  class ME2BertConfig(PretrainedConfig):
4
  model_type = "me2bert"
5
  architectures = ["ME2BertModel"]
 
 
 
 
6
 
7
  def __init__(
8
  self,
 
3
  class ME2BertConfig(PretrainedConfig):
4
  model_type = "me2bert"
5
  architectures = ["ME2BertModel"]
6
+ auto_map = {
7
+ "AutoConfig": "configuration_me2bert.ME2BertConfig",
8
+ "AutoModel": "modeling_me2bert.ME2BertModel",
9
+ }
10
 
11
  def __init__(
12
  self,