104 lines
3.0 KiB
TOML
104 lines
3.0 KiB
TOML
[package]
|
|
name = "candle-examples"
|
|
version.workspace = true
|
|
edition.workspace = true
|
|
description.workspace = true
|
|
repository.workspace = true
|
|
keywords.workspace = true
|
|
categories.workspace = true
|
|
license.workspace = true
|
|
readme = "README.md"
|
|
|
|
[dependencies]
|
|
accelerate-src = { workspace = true, optional = true }
|
|
candle = { workspace = true }
|
|
candle-datasets = { workspace = true, optional = true }
|
|
candle-nn = { workspace = true }
|
|
candle-transformers = { workspace = true }
|
|
candle-flash-attn = { workspace = true, optional = true }
|
|
candle-onnx = { workspace = true, optional = true }
|
|
|
|
csv = "1.3.0"
|
|
cudarc = { workspace = true, optional = true }
|
|
half = { workspace = true, optional = true }
|
|
hf-hub = { workspace = true, features = ["tokio"] }
|
|
image = { workspace = true }
|
|
intel-mkl-src = { workspace = true, optional = true }
|
|
num-traits = { workspace = true }
|
|
pyo3 = { version = "0.20.0", features = ["auto-initialize"], optional = true }
|
|
rayon = { workspace = true }
|
|
rubato = { version = "0.15.0", optional = true }
|
|
safetensors = { workspace = true }
|
|
serde = { workspace = true }
|
|
serde_json = { workspace = true }
|
|
symphonia = { version = "0.5.3", features = ["all"], optional = true }
|
|
tokenizers = { workspace = true, features = ["onig"] }
|
|
cpal= { version = "0.15.2", optional = true }
|
|
|
|
[dev-dependencies]
|
|
anyhow = { workspace = true }
|
|
byteorder = { workspace = true }
|
|
clap = { workspace = true }
|
|
imageproc = { workspace = true }
|
|
memmap2 = { workspace = true }
|
|
rand = { workspace = true }
|
|
ab_glyph = { workspace = true }
|
|
tracing = { workspace = true }
|
|
tracing-chrome = { workspace = true }
|
|
tracing-subscriber = { workspace = true }
|
|
# Necessary to disambiguate with tokio in wasm examples which are 1.28.1
|
|
tokio = "1.29.1"
|
|
|
|
[build-dependencies]
|
|
anyhow = { workspace = true }
|
|
bindgen_cuda = { version = "0.1.1", optional = true }
|
|
|
|
[features]
|
|
default = []
|
|
accelerate = ["dep:accelerate-src", "candle/accelerate", "candle-nn/accelerate", "candle-transformers/accelerate"]
|
|
cuda = ["candle/cuda", "candle-nn/cuda", "candle-transformers/cuda", "dep:bindgen_cuda"]
|
|
cudnn = ["candle/cudnn"]
|
|
flash-attn = ["cuda", "candle-transformers/flash-attn", "dep:candle-flash-attn"]
|
|
mkl = ["dep:intel-mkl-src", "candle/mkl", "candle-nn/mkl", "candle-transformers/mkl"]
|
|
nccl = ["cuda", "cudarc/nccl", "dep:half"]
|
|
onnx = ["candle-onnx"]
|
|
metal = ["candle/metal", "candle-nn/metal"]
|
|
microphone = ["cpal"]
|
|
encodec = ["cpal", "symphonia", "rubato"]
|
|
|
|
[[example]]
|
|
name = "llama_multiprocess"
|
|
required-features = ["cuda", "nccl", "flash-attn"]
|
|
|
|
[[example]]
|
|
name = "reinforcement-learning"
|
|
required-features = ["pyo3"]
|
|
|
|
[[example]]
|
|
name = "onnx"
|
|
required-features = ["onnx"]
|
|
|
|
[[example]]
|
|
name = "onnx_basics"
|
|
required-features = ["onnx"]
|
|
|
|
[[example]]
|
|
name = "whisper"
|
|
required-features = ["symphonia"]
|
|
|
|
[[example]]
|
|
name = "whisper-microphone"
|
|
required-features = ["microphone"]
|
|
|
|
[[example]]
|
|
name = "mnist-training"
|
|
required-features = ["candle-datasets"]
|
|
|
|
[[example]]
|
|
name = "llama2-c"
|
|
required-features = ["candle-datasets"]
|
|
|
|
[[example]]
|
|
name = "encodec"
|
|
required-features = ["encodec"]
|