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

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-transformers/Cargo.toml
View File

@ -31,3 +31,4 @@ accelerate = ["dep:accelerate-src", "candle/accelerate", "candle-nn/accelerate"]
cuda = ["candle/cuda", "candle-nn/cuda"]
flash-attn = ["cuda", "dep:candle-flash-attn"]
mkl = ["dep:intel-mkl-src", "candle/mkl", "candle-nn/mkl"]
metal = ["candle/metal", "candle-nn/metal"]