[package]
edition = "2024"
name = "infraqueue-language-model"
version = "0.1.0"
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Language model generation for INFRAQUEUE AI"
homepage = "https://github.com/helloimalemur/infraqueue"
readme = false
license = "MIT"
repository = "https://github.com/helloimalemur/infraqueue"
resolver = "2"
[features]
accelerate = [
"dep:accelerate-src",
"candle-core/accelerate",
"candle-nn/accelerate",
"candle-examples/accelerate",
"candle-transformers/accelerate",
]
cuda = [
"candle-core/cuda",
"candle-nn/cuda",
"candle-examples/cuda",
"candle-transformers/cuda",
]
default = []
metal = [
"candle-core/metal",
"candle-nn/metal",
"candle-examples/metal",
"candle-transformers/metal",
]
mkl = [
"dep:intel-mkl-src",
"candle-core/mkl",
"candle-nn/mkl",
"candle-examples/mkl",
"candle-transformers/mkl",
]
[lib]
name = "infraqueue_language_model"
path = "src/lib.rs"
[[bin]]
name = "infraqueue-language-model"
path = "src/main.rs"
[dependencies.accelerate-src]
version = "0.3"
optional = true
[dependencies.anyhow]
version = "1.0.98"
[dependencies.candle-core]
version = "0.9.1"
[dependencies.candle-examples]
version = "0.9.1"
[dependencies.candle-nn]
version = "0.9.1"
[dependencies.candle-transformers]
version = "0.9.1"
[dependencies.clap]
version = "4.5.41"
features = ["derive"]
[dependencies.hf-hub]
version = "0.4.3"
default-features = true
package = "hf-hub"
[dependencies.intel-mkl-src]
version = "0.8"
optional = true
[dependencies.reqwest]
version = "0.12"
features = [
"blocking",
"json",
"rustls-tls",
]
default-features = false
[dependencies.serde]
version = "1.0"
features = ["derive"]
[dependencies.serde_json]
version = "1.0.140"
[dependencies.tokenizers]
version = "0.21.4"
[dependencies.tracing-chrome]
version = "0.7.2"
[dependencies.tracing-subscriber]
version = "0.3.19"