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

Merge pull request #1318 from huggingface/metal4

Browse Source 
Starting to fix some tests.
This commit is contained in:
Nicolas Patry
2023-12-20 15:37:31 +01:00
committed by GitHub
parent 96f1a28e39 9b5e4843a6
commit 9fc210fae8
25 changed files with 2784 additions and 785 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-examples/Cargo.toml
View File

@ -57,6 +57,7 @@ flash-attn = ["cuda", "candle-transformers/flash-attn", "dep:candle-flash-attn"]
mkl = ["dep:intel-mkl-src", "candle/mkl", "candle-nn/mkl", "candle-transformers/mkl"]
nccl = ["cuda", "cudarc/nccl", "dep:half"]
onnx = ["candle-onnx"]
metal = ["candle/metal", "candle-nn/metal"]
[[example]]
name = "llama_multiprocess"