CodeGeeX4/candle_demo/Cargo.toml

37 lines
1.1 KiB
TOML
Raw Normal View History

2024-07-11 15:31:59 +00:00
[package]
name = "codegeex4-candle"
version = "0.1.0"
edition = "2021"
authors = ["Donjuan Platinum <donjuan@lecturify.net>"]
license = "GPL-2.0-only"
description = "Codegeex4"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
# candle-transformers = {path = "../candle/candle-transformers"}
# candle-core = {path = "../candle/candle-core"}
# candle-nn = {path = "../candle/candle-nn"}
2024-07-12 16:22:25 +00:00
#anyhow = "1.0.86"
2024-07-11 15:31:59 +00:00
hf-hub = "0.3.2"
#tokenizer = "0.1.2"
clap = { version = "4.5.6", features = ["derive"] }
#tracing-chrome = "0.7.2"
#candle-examples = {path = "../candle/candle-examples"}
#tracing-subscriber = "0.3.18"
tokenizers = "0.19.1"
serde_json = "1.0.120"
candle-core = "0.6.0"
candle-transformers = "0.6.0"
candle-examples = "0.6.0"
candle-nn = "0.6.0"
safetensors = "0.4.3"
2024-07-12 16:22:25 +00:00
#safetensors = {path ="../safetensors/safetensors"}
2024-07-14 11:11:31 +00:00
[build-dependencies]
bindgen_cuda = { version = "0.1.1", optional = true }
2024-07-14 11:10:33 +00:00
[features]
default = []
cuda = ["candle/cuda", "candle-nn/cuda", "candle-transformers/cuda", "dep:bindgen_cuda"]