diff options
Diffstat (limited to 'candle-transformers')
-rw-r--r-- | candle-transformers/Cargo.toml | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-transformers/Cargo.toml b/candle-transformers/Cargo.toml index 37e5fa65..000702f9 100644 --- a/candle-transformers/Cargo.toml +++ b/candle-transformers/Cargo.toml @@ -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"] |