flash_rerank 0.2.0

Core reranking engine — cross-encoder and ColBERT inference via ONNX Runtime
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

[package]
name = "flash_rerank"
version.workspace = true
edition.workspace = true
rust-version.workspace = true
license.workspace = true
repository.workspace = true
description = "Core reranking engine — cross-encoder and ColBERT inference via ONNX Runtime"
keywords.workspace = true
categories.workspace = true
authors.workspace = true

[dependencies]
ort = { version = "2.0.0-rc.12" }
tokenizers = { version = "0.21", default-features = false, features = ["fancy-regex"] }
memmap2 = "0.9"
hf-hub = { version = "0.4", features = ["tokio"] }
serde = { workspace = true }
serde_json = { workspace = true }
tokio = { workspace = true }
tracing = { workspace = true }
thiserror = { workspace = true }
indicatif = "0.17"
lru = "0.12"

[[example]]
name = "profile_pipeline"

[[example]]
name = "micro_profile"

[[example]]
name = "gpu_check"

[[example]]
name = "pipeline_benchmark"

[[example]]
name = "gpu_bench_models"

[features]
default = []
cuda = []
tensorrt = []

[dev-dependencies]
insta = { version = "1", features = ["yaml"] }
proptest = "1"
tokio = { version = "1", features = ["full"] }
anyhow = "1"
dirs = "6"