62 lines
1.7 KiB
TOML
62 lines
1.7 KiB
TOML
[package]
|
|
name = "input_classifier"
|
|
authors = ["Warp Team <dev@warp.dev>"]
|
|
edition = "2024"
|
|
publish.workspace = true
|
|
license.workspace = true
|
|
|
|
[[bin]]
|
|
name = "evaluate"
|
|
path = "src/bin/evaluate.rs"
|
|
|
|
[features]
|
|
onnx_candle = [
|
|
"onnx",
|
|
"dep:candle-core",
|
|
"dep:candle-nn",
|
|
"dep:candle-onnx",
|
|
"dep:prost",
|
|
]
|
|
onnx_ort = [
|
|
"onnx",
|
|
"dep:ort",
|
|
"dep:ndarray",
|
|
"dep:parking_lot",
|
|
]
|
|
onnx = ["dep:rust-embed", "dep:tokenizers"]
|
|
# Selects which BERT-tiny ONNX model bytes are embedded via rust-embed; enable exactly one.
|
|
nld_classifier_v1 = ["onnx_candle"]
|
|
nld_classifier_v2 = ["onnx_candle"]
|
|
nld_classifier_v3 = ["onnx_candle"]
|
|
# Selects which shell-command heuristic behavior is used. If both are enabled, v2 wins.
|
|
nld_heuristic_v1 = []
|
|
nld_heuristic_v2 = []
|
|
|
|
[dependencies]
|
|
anyhow.workspace = true
|
|
async-trait.workspace = true
|
|
clap.workspace = true
|
|
futures.workspace = true
|
|
candle-core = { version = "0.9.2", optional = true }
|
|
candle-nn = { version = "0.9.2", optional = true }
|
|
candle-onnx = { version = "0.9.2", optional = true }
|
|
cfg-if.workspace = true
|
|
itertools.workspace = true
|
|
futures-lite.workspace = true
|
|
instant.workspace = true
|
|
lazy_static.workspace = true
|
|
log.workspace = true
|
|
natural_language_detection.workspace = true
|
|
ndarray = { version = "0.16.1", optional = true }
|
|
ort = { version = "2.0.0-rc.10", optional = true }
|
|
parking_lot = { workspace = true, optional = true }
|
|
prost = { version = "0.14.3", optional = true }
|
|
rust-embed = { workspace = true, optional = true }
|
|
serde.workspace = true
|
|
smol_str.workspace = true
|
|
tokenizers = { version = "0.21.4", optional = true }
|
|
galaxy_completer.workspace = true
|
|
|
|
[dev-dependencies]
|
|
warp_features.workspace = true
|