mxbai-edge-colbert-v0-17m / onnx_config.json
bclavie's picture
fix: Include all Dense projection layers in ONNX export (output dim 48) (#3)
21996dc
Raw
History Blame Contribute Delete
803 Bytes
{
"model_type": "ColBERT",
"model_name": "mixedbread-ai/mxbai-edge-colbert-v0-17m",
"model_class": "ModernBertModel",
"uses_token_type_ids": false,
"query_prefix": "[Q] ",
"document_prefix": "[D] ",
"query_length": 48,
"document_length": 512,
"do_query_expansion": false,
"attend_to_expansion_tokens": false,
"skiplist_words": [
"!",
"\"",
"#",
"$",
"%",
"&",
"'",
"(",
")",
"*",
"+",
",",
"-",
".",
"/",
":",
";",
"<",
"=",
">",
"?",
"@",
"[",
"\\",
"]",
"^",
"_",
"`",
"{",
"|",
"}",
"~"
],
"embedding_dim": 48,
"mask_token_id": 50284,
"pad_token_id": 50284,
"query_prefix_id": 50368,
"document_prefix_id": 50369,
"do_lower_case": true
}