huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 18:28:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Centralize the dependency versions and inherit them. (#177)

Browse Source
This commit is contained in:
Laurent Mazare
2023-07-16 07:47:17 +01:00
committed by GitHub
parent 3fb1c4ea96
commit 104f89df31
8 changed files with 91 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
candle-core/Cargo.toml
View File

@ -11,27 +11,23 @@ license = "MIT/Apache-2.0"
readme = "README.md"
[dependencies]
byteorder = "1.4.3"
byteorder = { workspace = true }
candle-kernels = { path = "../candle-kernels", optional = true }
# Re-enable this once 0.9.13 as been released as it would include the cublas-f16 changes
# cudarc = { version = "0.9.13", optional = true, features = ["f16"] }
cudarc = { git = "https://github.com/LaurentMazare/cudarc.git", branch = "cublas-bf16", optional = true, features = ["f16"] }
# TODO: Switch back to the official gemm implementation once something similar to
# https://github.com/sarah-ek/gemm/pull/8 is available.
gemm = { git = "https://github.com/LaurentMazare/gemm.git", branch = "f16-vectorize-pack" }
half = { version = "2.3.1", features = ["num-traits"] }
intel-mkl-src = {version="0.8.1", optional=true, features = ["mkl-dynamic-lp64-iomp"]}
libc = { version = "0.2.147", optional = true }
memmap2 = "0.7.1"
num-traits = "0.2.15"
num_cpus = "1.15.0"
rand = "0.8.5"
safetensors = "0.3.1"
thiserror = "1"
zip = { version = "0.6.6", default-features=false }
cudarc = { workspace = true, optional = true }
gemm = { workspace = true }
half = { workspace = true }
intel-mkl-src = { workspace = true, optional = true }
libc = { workspace = true, optional = true }
memmap2 = { workspace = true }
num-traits = { workspace = true }
num_cpus = { workspace = true }
rand = { workspace = true }
safetensors = { workspace = true }
thiserror = { workspace = true }
zip = { workspace = true }
[dev-dependencies]
anyhow = { version = "1", features = ["backtrace"] }
anyhow = { workspace = true }
[features]
default = []