Use matryoshka_dimensions to control the allowed output dimensions. (#13)
Browse files- Use matryoshka_dimensions to control the allowed output dimensions. (d149fe28b43e38f1ddef5f4f4b120b44eac1506e)
Co-authored-by: noooop <[email protected]>
- config.json +1 -0
config.json
CHANGED
|
@@ -24,5 +24,6 @@
|
|
| 24 |
"transformers_version": "4.39.3",
|
| 25 |
"type_vocab_size": 1,
|
| 26 |
"use_cache": true,
|
|
|
|
| 27 |
"vocab_size": 250002
|
| 28 |
}
|
|
|
|
| 24 |
"transformers_version": "4.39.3",
|
| 25 |
"type_vocab_size": 1,
|
| 26 |
"use_cache": true,
|
| 27 |
+
"matryoshka_dimensions": [256],
|
| 28 |
"vocab_size": 250002
|
| 29 |
}
|