From 9b36e9056113e7dd1e516adb0283dc4c8f8b5eb4 Mon Sep 17 00:00:00 2001 From: donjuanplatinum <113148619+donjuanplatinum@users.noreply.github.com> Date: Sun, 14 Jul 2024 19:10:33 +0800 Subject: [PATCH] Update Cargo.toml --- candle_demo/Cargo.toml | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/candle_demo/Cargo.toml b/candle_demo/Cargo.toml index 0d80876..3126d88 100755 --- a/candle_demo/Cargo.toml +++ b/candle_demo/Cargo.toml @@ -26,3 +26,8 @@ candle-examples = "0.6.0" candle-nn = "0.6.0" safetensors = "0.4.3" #safetensors = {path ="../safetensors/safetensors"} + +[features] +default = [] +cuda = ["candle/cuda", "candle-nn/cuda", "candle-transformers/cuda", "dep:bindgen_cuda"] +