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

More f16 and bf16 support.

Browse Source
This commit is contained in:
laurent
2023-06-26 20:52:01 +01:00
parent a31411fd91
commit 22da2c7e02
3 changed files with 392 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Cargo.toml
View File

@ -23,7 +23,7 @@ candle-kernels = { path = "kernels", optional = true }
gemm = "0.15.4"
zip = { version = "0.6.6", default-features=false }
byteorder = "1.4.3"
half = "2.3.1"
half = { version = "2.3.1", features = ["num-traits"] }
num-traits = "0.2.15"
[dev-dependencies]
@ -33,5 +33,5 @@ rand = "0.8.5"
tokenizers = { version = "0.13.3", default-features=false, features=["onig"] }
[features]
default = []
default = ["cuda"]
cuda = ["dep:cudarc", "dep:candle-kernels"]