From 9fc0744360207b0a428cc435035b106ee6c22076 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gabriel=20Mart=C3=ADn=20Bl=C3=A1zquez?= Date: Thu, 5 Sep 2024 09:51:41 +0200 Subject: [PATCH] Add `candle-nn` deps --- candle-holder-models/Cargo.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/candle-holder-models/Cargo.toml b/candle-holder-models/Cargo.toml index c45c5e3..76f1181 100644 --- 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"]