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: Added benchmarks #19

Closed joshniemela closed 8 months ago

joshniemela commented 9 months ago

There is a lot of parallelisation in the embedding functions which I suspect some of them may just be adding additional overhead, this benchmark should serve as a first step to optimising the performance further.

Anush008 commented 9 months ago

@joshniemela, thanks for scaffolding this.

Some notes:

Anush008 commented 9 months ago

Also, please pull the latest changes from the upstream. 5bfd2de6d2139d074a18d084c8655c4ad78a7ea4, for the tests to pass.

joshniemela commented 8 months ago

Removed version bump and fixed the two linting errors, sorry for taking so long I had some exams to do.

Anush008 commented 8 months ago

Thanks for this @joshniemela. I am thinking of adding https://codspeed.io/ on top of this. As it goes well with Criterion.

github-actions[bot] commented 8 months ago

:tada: This PR is included in version 2.1.0 :tada:

The release is available on:

Your semantic-release bot :package::rocket: