mirror of
https://github.com/huggingface/candle.git
synced 2025-06-14 09:57:10 +00:00
Refactor the hierarchy.
This commit is contained in:
36
Cargo.toml
36
Cargo.toml
@ -1,37 +1,7 @@
|
||||
[package]
|
||||
name = "candle"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
description = "Minimalist ML framework."
|
||||
repository = "https://github.com/LaurentMazare/candle"
|
||||
keywords = ["blas", "tensor", "machine-learning"]
|
||||
categories = ["science"]
|
||||
license = "MIT/Apache-2.0"
|
||||
readme = "README.md"
|
||||
|
||||
[workspace]
|
||||
|
||||
members = [
|
||||
"kernels",
|
||||
"candle-core",
|
||||
"candle-kernels",
|
||||
]
|
||||
|
||||
[dependencies]
|
||||
safetensors = "0.3.1"
|
||||
thiserror = "1"
|
||||
cudarc = { version = "0.9.9", optional = true, features = ["f16"] }
|
||||
candle-kernels = { path = "kernels", optional = true }
|
||||
gemm = "0.15.4"
|
||||
zip = { version = "0.6.6", default-features=false }
|
||||
byteorder = "1.4.3"
|
||||
half = { version = "2.3.1", features = ["num-traits"] }
|
||||
num-traits = "0.2.15"
|
||||
|
||||
[dev-dependencies]
|
||||
anyhow = "1"
|
||||
clap = { version = "4.2.4", features = ["derive"] }
|
||||
rand = "0.8.5"
|
||||
tokenizers = { version = "0.13.3", default-features=false, features=["onig"] }
|
||||
|
||||
[features]
|
||||
default = ["cuda"]
|
||||
cuda = ["dep:cudarc", "dep:candle-kernels"]
|
||||
|
32
candle-core/Cargo.toml
Normal file
32
candle-core/Cargo.toml
Normal file
@ -0,0 +1,32 @@
|
||||
[package]
|
||||
name = "candle"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
description = "Minimalist ML framework."
|
||||
repository = "https://github.com/LaurentMazare/candle"
|
||||
keywords = ["blas", "tensor", "machine-learning"]
|
||||
categories = ["science"]
|
||||
license = "MIT/Apache-2.0"
|
||||
readme = "README.md"
|
||||
|
||||
[dependencies]
|
||||
safetensors = "0.3.1"
|
||||
thiserror = "1"
|
||||
cudarc = { version = "0.9.9", optional = true, features = ["f16"] }
|
||||
candle-kernels = { path = "../candle-kernels", optional = true }
|
||||
gemm = "0.15.4"
|
||||
zip = { version = "0.6.6", default-features=false }
|
||||
byteorder = "1.4.3"
|
||||
half = { version = "2.3.1", features = ["num-traits"] }
|
||||
num-traits = "0.2.15"
|
||||
|
||||
[dev-dependencies]
|
||||
anyhow = "1"
|
||||
clap = { version = "4.2.4", features = ["derive"] }
|
||||
rand = "0.8.5"
|
||||
tokenizers = { version = "0.13.3", default-features=false, features=["onig"] }
|
||||
|
||||
[features]
|
||||
default = ["cuda"]
|
||||
cuda = ["dep:cudarc", "dep:candle-kernels"]
|
Reference in New Issue
Block a user