Add candle-nn deps

gabrielmbmb · Sep 5, 2024 · 9fc0744 · 9fc0744
1 parent 903e97c
commit 9fc0744
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/candle-holder-models/Cargo.toml b/candle-holder-models/Cargo.toml
@@ -25,6 +25,6 @@ tokenizers = ["candle-holder-tokenizers"]
 metal = ["candle-core/metal", "candle-nn/metal", "candle-metal-kernels"]
 cuda = ["candle-core/cuda", "candle-nn/cuda"]
 cudnn = ["candle-core/cudnn"]
-accelerate = ["candle-core/accelerate"]
-mkl = ["candle-core/mkl"]
+accelerate = ["candle-core/accelerate", "candle-nn/accelerate"]
+mkl = ["candle-core/mkl", "candle-nn/mkl"]
 flash-attn = ["candle-flash-attn"]