Update configuration_me2bert.py
Browse files- 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,
|