Tamás Ficsor
commited on
Commit
•
e54cfb7
1
Parent(s):
dd6c160
add model
Browse files- config.json +1 -8
- config.py +10 -0
- pytorch_model.bin +2 -2
config.json
CHANGED
@@ -1,15 +1,8 @@
|
|
1 |
{
|
2 |
"architectures": [
|
3 |
-
"
|
4 |
],
|
5 |
"attention_probs_dropout_prob": 0.1,
|
6 |
-
"auto_map": {
|
7 |
-
"AutoConfig": "config.CharmenElectraConfig",
|
8 |
-
"AutoModel": "modeling_charmen.CharmenElectraModel",
|
9 |
-
"AutoModelForMultipleChoice": "modeling_charmen.CharmenElectraForMultipleChoice",
|
10 |
-
"AutoModelForSequenceClassification": "modeling_charmen.CharmenElectraForSequenceClassification",
|
11 |
-
"AutoModelForTokenClassification": "modeling_charmen.CharmenElectraForTokenClassification"
|
12 |
-
},
|
13 |
"classifier_dropout": null,
|
14 |
"downsampling_factor": 4,
|
15 |
"embedding_size": 768,
|
|
|
1 |
{
|
2 |
"architectures": [
|
3 |
+
"CharmenElectraModel"
|
4 |
],
|
5 |
"attention_probs_dropout_prob": 0.1,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
"classifier_dropout": null,
|
7 |
"downsampling_factor": 4,
|
8 |
"embedding_size": 768,
|
config.py
CHANGED
@@ -8,6 +8,16 @@ _SAMPLING_TYPE = Literal['fp32_gumbel', 'fp16_gumbel', 'multinomial']
|
|
8 |
class CharmenElectraConfig(PretrainedConfig):
|
9 |
model_type = "SzegedAI/charmen-electra"
|
10 |
_name_or_path = "SzegedAI/charmen-electra"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
|
12 |
def __init__(
|
13 |
self,
|
|
|
8 |
class CharmenElectraConfig(PretrainedConfig):
|
9 |
model_type = "SzegedAI/charmen-electra"
|
10 |
_name_or_path = "SzegedAI/charmen-electra"
|
11 |
+
architectures = [
|
12 |
+
"CharmenElectraModel"
|
13 |
+
]
|
14 |
+
auto_map = {
|
15 |
+
"AutoConfig": "config.CharmenElectraConfig",
|
16 |
+
"AutoModel": "modeling_charmen.CharmenElectraModel",
|
17 |
+
"AutoModelForMultipleChoice": "modeling_charmen.CharmenElectraForMultipleChoice",
|
18 |
+
"AutoModelForSequenceClassification": "modeling_charmen.CharmenElectraForSequenceClassification",
|
19 |
+
"AutoModelForTokenClassification": "modeling_charmen.CharmenElectraForTokenClassification"
|
20 |
+
}
|
21 |
|
22 |
def __init__(
|
23 |
self,
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7074667cdc918bf66a2b408b6e879995964891452d4dd598f0b42fbbdc0ee60b
|
3 |
+
size 173978597
|