oweller2
commited on
Commit
•
358a4d5
1
Parent(s):
26eb98d
add back in
Browse files- config.json +1 -0
- configuration_bert.py +1 -0
config.json
CHANGED
|
@@ -5,6 +5,7 @@
|
|
| 5 |
"FlexBertForCausalLM"
|
| 6 |
],
|
| 7 |
"auto_map": {
|
|
|
|
| 8 |
"AutoModel": "orionweller/test-flex-gpt--modeling_flexbert.FlexBertModel",
|
| 9 |
"AutoModelForCausalLM": "orionweller/test-flex-gpt--modeling_flexbert.FlexBertForCausalLM"
|
| 10 |
},
|
|
|
|
| 5 |
"FlexBertForCausalLM"
|
| 6 |
],
|
| 7 |
"auto_map": {
|
| 8 |
+
"AutoConfig": "configuration_bert.FlexBertConfig",
|
| 9 |
"AutoModel": "orionweller/test-flex-gpt--modeling_flexbert.FlexBertModel",
|
| 10 |
"AutoModelForCausalLM": "orionweller/test-flex-gpt--modeling_flexbert.FlexBertForCausalLM"
|
| 11 |
},
|
configuration_bert.py
CHANGED
|
@@ -40,6 +40,7 @@ class BertConfig(TransformersBertConfig):
|
|
| 40 |
|
| 41 |
|
| 42 |
class FlexBertConfig(TransformersBertConfig):
|
|
|
|
| 43 |
def __init__(
|
| 44 |
self,
|
| 45 |
attention_layer: str = "base",
|
|
|
|
| 40 |
|
| 41 |
|
| 42 |
class FlexBertConfig(TransformersBertConfig):
|
| 43 |
+
model_type = "flex_bert"
|
| 44 |
def __init__(
|
| 45 |
self,
|
| 46 |
attention_layer: str = "base",
|