Anush008 / fastembed-rs

Library for generating vector embeddings, reranking in Rust
https://docs.rs/fastembed
Apache License 2.0
264 stars 36 forks source link

feat: mean pooling #99

Closed triandco closed 2 months ago

triandco commented 2 months ago

The current implementation does not include pooling. This result in wildly different embedding to fastembed python counterpart.

This pull request add a default mean pooling for token embeddings output from ort session.

This pull request includes:

denwong47 commented 2 months ago

Noted, will take a look after work.

github-actions[bot] commented 1 month ago

:tada: This issue has been resolved in version 4.0.0 :tada:

The release is available on:

Your semantic-release bot :package::rocket: