Use matryoshka_dimensions to control the allowed output dimensions. (#13)
Browse files- Use matryoshka_dimensions to control the allowed output dimensions. (c6f9424b4d35ad6d408687d179d568bd68094afd)
Co-authored-by: noooop <[email protected]>
- config.json +1 -0
config.json
CHANGED
|
@@ -31,5 +31,6 @@
|
|
| 31 |
"type_vocab_size": 1,
|
| 32 |
"unpad_inputs": "true",
|
| 33 |
"use_memory_efficient_attention": "true",
|
|
|
|
| 34 |
"vocab_size": 250048
|
| 35 |
}
|
|
|
|
| 31 |
"type_vocab_size": 1,
|
| 32 |
"unpad_inputs": "true",
|
| 33 |
"use_memory_efficient_attention": "true",
|
| 34 |
+
"matryoshka_dimensions": [256],
|
| 35 |
"vocab_size": 250048
|
| 36 |
}
|