avsolatorio's picture
Fix the computation of the mean embedding
b387470
raw
history blame contribute delete
331 Bytes
{
"word_embedding_dimension": 384,
"pooling_mode_cls_token": false,
"pooling_mode_mean_tokens": true,
"pooling_mode_max_tokens": false,
"pooling_mode_mean_sqrt_len_tokens": false,
"pooling_mode_weightedmean_tokens": false,
"pooling_mode_lasttoken": false,
"include_prompt": true,
"output_key": "query_embedding"
}