bwang0911's picture
feat: add pooling config for sbert (#7)
ac0b6c5 verified
raw
history blame
192 Bytes
{
"word_embedding_dimension": 1024,
"pooling_mode_cls_token": false,
"pooling_mode_mean_tokens": true,
"pooling_mode_max_tokens": false,
"pooling_mode_mean_sqrt_len_tokens": false
}