[package]
edition = "2021"
name = "dakera-inference"
version = "0.6.2"
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Embedded inference engine for Dakera - generates embeddings locally"
readme = false
license = "MIT"
repository = "https://github.com/Dakera-AI/dakera"
[features]
cpu = []
cuda = [
"candle-core/cuda",
"candle-nn/cuda",
"candle-transformers/cuda",
]
default = ["cpu"]
metal = [
"candle-core/metal",
"candle-nn/metal",
"candle-transformers/metal",
]
[lib]
name = "inference"
path = "src/lib.rs"
[dependencies.async-trait]
version = "0.1"
[dependencies.candle-core]
version = "0.8"
[dependencies.candle-nn]
version = "0.8"
[dependencies.candle-transformers]
version = "0.8"
[dependencies.common]
version = "0.6.0"
package = "dakera-common"
[dependencies.hf-hub]
version = "0.3"
features = ["tokio"]
[dependencies.parking_lot]
version = "0.12"
[dependencies.serde]
version = "1.0"
features = ["derive"]
[dependencies.serde_json]
version = "1.0"
[dependencies.thiserror]
version = "1.0"
[dependencies.tokenizers]
version = "0.20"
features = ["onig"]
default-features = false
[dependencies.tokio]
version = "1.37"
features = ["full"]
[dependencies.tracing]
version = "0.1"
[dependencies.ureq]
version = "2"
features = [
"tls",
"native-certs",
]
default-features = false
[dependencies.url]
version = "2"
[dev-dependencies.tokio]
version = "1.37"
features = [
"full",
"rt-multi-thread",
"macros",
]