mirror of
https://github.com/geoffsee/predict-otron-9001.git
synced 2025-09-08 22:46:44 +00:00
29 lines
1.3 KiB
TOML
29 lines
1.3 KiB
TOML
[package]
|
|
name = "llama-runner"
|
|
version = "0.1.0"
|
|
edition = "2021"
|
|
|
|
[dependencies]
|
|
candle-core = { git = "https://github.com/huggingface/candle.git" }
|
|
candle-nn = { git = "https://github.com/huggingface/candle.git" }
|
|
candle-transformers = { git = "https://github.com/huggingface/candle.git" }
|
|
hf-hub = "0.3"
|
|
tokenizers = "0.20"
|
|
anyhow = "1.0"
|
|
clap = { version = "4.0", features = ["derive", "string"] }
|
|
serde_json = "1.0"
|
|
|
|
[target.'cfg(target_os = "macos")'.dependencies]
|
|
candle-core = { git = "https://github.com/huggingface/candle.git", features = ["metal"] }
|
|
candle-nn = { git = "https://github.com/huggingface/candle.git", features = ["metal"] }
|
|
candle-transformers = { git = "https://github.com/huggingface/candle.git", features = ["metal"] }
|
|
|
|
[target.'cfg(not(target_os = "macos"))'.dependencies]
|
|
candle-core = { git = "https://github.com/huggingface/candle.git", features = ["cuda"], optional = true }
|
|
candle-nn = { git = "https://github.com/huggingface/candle.git", features = ["cuda"], optional = true }
|
|
candle-transformers = { git = "https://github.com/huggingface/candle.git", features = ["cuda"], optional = true }
|
|
|
|
[features]
|
|
default = []
|
|
cuda = ["candle-core/cuda", "candle-nn/cuda", "candle-transformers/cuda"]
|
|
metal = ["candle-core/metal", "candle-nn/metal", "candle-transformers/metal"] |