diff --git a/backend/vector_store/Cargo.toml b/backend/vector_store/Cargo.toml index 3e53e14..9616365 100644 --- a/backend/vector_store/Cargo.toml +++ b/backend/vector_store/Cargo.toml @@ -31,8 +31,3 @@ futures-core = "0.3.30" #codegen-units = 1 #panic = "abort" #debug=true - - -[[bench]] -name = "ort" -harness = false diff --git a/backend/vector_store/benches/ort.rs b/backend/vector_store/benches/ort.rs deleted file mode 100644 index 2af7f9f..0000000 --- a/backend/vector_store/benches/ort.rs +++ /dev/null @@ -1,18 +0,0 @@ -use criterion::{criterion_group, criterion_main, Criterion}; - -use fastembed::{EmbeddingBase, EmbeddingModel, FlagEmbedding, InitOptions}; - -fn criterion_benchmark(c: &mut Criterion) { - let model: FlagEmbedding = FlagEmbedding::try_new(InitOptions { - model_name: EmbeddingModel::AllMiniLML6V2, - show_download_message: false, - ..Default::default() - }) - .unwrap(); - c.bench_function("ortv2", |b| { - b.iter(|| model.query_embed("Hello, World!").unwrap()) - }); -} - -criterion_group!(benches, criterion_benchmark); -criterion_main!(benches);