update state_dict for general mlm loading
Browse files- config.json +3 -5
- model.safetensors +2 -2
config.json
CHANGED
|
@@ -1,12 +1,9 @@
|
|
| 1 |
{
|
| 2 |
-
"_name_or_path": "bert-base-uncased",
|
| 3 |
"add_marker_tokens": false,
|
| 4 |
"architectures": [
|
| 5 |
"SpladeBertModel"
|
| 6 |
],
|
| 7 |
-
"attend_to_doc_expanded_tokens":
|
| 8 |
-
false
|
| 9 |
-
],
|
| 10 |
"attend_to_query_expanded_tokens": false,
|
| 11 |
"attention_probs_dropout_prob": 0.1,
|
| 12 |
"backbone_model_type": "bert",
|
|
@@ -37,8 +34,9 @@
|
|
| 37 |
"save_step": 49009,
|
| 38 |
"similarity_function": "dot",
|
| 39 |
"sparsification": "relu_log",
|
|
|
|
| 40 |
"torch_dtype": "float32",
|
| 41 |
-
"transformers_version": "4.
|
| 42 |
"type_vocab_size": 2,
|
| 43 |
"use_cache": true,
|
| 44 |
"vocab_size": 30522
|
|
|
|
| 1 |
{
|
|
|
|
| 2 |
"add_marker_tokens": false,
|
| 3 |
"architectures": [
|
| 4 |
"SpladeBertModel"
|
| 5 |
],
|
| 6 |
+
"attend_to_doc_expanded_tokens": false,
|
|
|
|
|
|
|
| 7 |
"attend_to_query_expanded_tokens": false,
|
| 8 |
"attention_probs_dropout_prob": 0.1,
|
| 9 |
"backbone_model_type": "bert",
|
|
|
|
| 34 |
"save_step": 49009,
|
| 35 |
"similarity_function": "dot",
|
| 36 |
"sparsification": "relu_log",
|
| 37 |
+
"tie_projection": true,
|
| 38 |
"torch_dtype": "float32",
|
| 39 |
+
"transformers_version": "4.50.2",
|
| 40 |
"type_vocab_size": 2,
|
| 41 |
"use_cache": true,
|
| 42 |
"vocab_size": 30522
|
model.safetensors
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:aa16ab7766c00e0c173f08cad5e703d800f535d575ae9f43eeb297706cea72da
|
| 3 |
+
size 438079944
|