oweller2
commited on
Commit
•
0f166fe
1
Parent(s):
27bdfe2
unpad
Browse files- config.json +2 -2
config.json
CHANGED
@@ -5,7 +5,7 @@
|
|
5 |
"FlexBertForCausalLM"
|
6 |
],
|
7 |
"auto_map": {
|
8 |
-
"AutoConfig": "
|
9 |
"AutoModel": "orionweller/test-flex-gpt--modeling_flexbert.FlexBertModel",
|
10 |
"AutoModelForCausalLM": "orionweller/test-flex-gpt--modeling_flexbert.FlexBertForCausalLM"
|
11 |
},
|
@@ -87,5 +87,5 @@
|
|
87 |
"use_fa2": true,
|
88 |
"use_sdpa_attn_mask": false,
|
89 |
"vocab_size": 50368,
|
90 |
-
"
|
91 |
}
|
|
|
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 |
},
|
|
|
87 |
"use_fa2": true,
|
88 |
"use_sdpa_attn_mask": false,
|
89 |
"vocab_size": 50368,
|
90 |
+
"is_causal": true
|
91 |
}
|