model-test-onnx / onnx_config.json
raphaelsty's picture
Upload onnx_config.json with huggingface_hub
1df8375 verified
{
"model_type": "ColBERT",
"model_name": "answerdotai/answerai-colbert-small-v1",
"model_class": "BertModel",
"uses_token_type_ids": true,
"query_prefix": "[unused0]",
"document_prefix": "[unused1]",
"query_length": 32,
"document_length": 300,
"do_query_expansion": false,
"attend_to_expansion_tokens": false,
"skiplist_words": [
"!",
"\"",
"#",
"$",
"%",
"&",
"'",
"(",
")",
"*",
"+",
",",
"-",
".",
"/",
":",
";",
"<",
"=",
">",
"?",
"@",
"[",
"\\",
"]",
"^",
"_",
"`",
"{",
"|",
"}",
"~"
],
"embedding_dim": 96,
"mask_token_id": 103,
"pad_token_id": 103,
"query_prefix_id": 1,
"document_prefix_id": 2,
"do_lower_case": false
}