Piekarski, Patryk (ext) commited on
Commit ·
985e106
1
Parent(s): 78d50a1
Add Pooling module for TEI
Browse files- 1_Pooling/config.json +10 -0
- modules.json +3 -12
1_Pooling/config.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"word_embedding_dimension": 768,
|
| 3 |
+
"pooling_mode_cls_token": false,
|
| 4 |
+
"pooling_mode_mean_tokens": true,
|
| 5 |
+
"pooling_mode_max_tokens": false,
|
| 6 |
+
"pooling_mode_mean_sqrt_len_tokens": false,
|
| 7 |
+
"pooling_mode_weightedmean_tokens": false,
|
| 8 |
+
"pooling_mode_lasttoken": false,
|
| 9 |
+
"include_prompt": false
|
| 10 |
+
}
|
modules.json
CHANGED
|
@@ -1,14 +1,5 @@
|
|
| 1 |
[
|
| 2 |
-
{
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
"path": "",
|
| 6 |
-
"type": "sentence_transformers.models.Transformer"
|
| 7 |
-
},
|
| 8 |
-
{
|
| 9 |
-
"idx": 1,
|
| 10 |
-
"name": "1",
|
| 11 |
-
"path": "1_Dense",
|
| 12 |
-
"type": "pylate.models.Dense.Dense"
|
| 13 |
-
}
|
| 14 |
]
|
|
|
|
| 1 |
[
|
| 2 |
+
{ "idx": 0, "name": "0", "path": "", "type": "sentence_transformers.models.Transformer" },
|
| 3 |
+
{ "idx": 1, "name": "1", "path": "1_Pooling", "type": "sentence_transformers.models.Pooling" },
|
| 4 |
+
{ "idx": 2, "name": "2", "path": "2_Dense", "type": "sentence_transformers.models.Dense" }
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
]
|