donjuanplatinum 1 год назад
Родитель
Сommit
9b36e90561
1 измененных файлов с 5 добавлено и 0 удалено
  1. 5 0
      candle_demo/Cargo.toml

+ 5 - 0
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"]
+