Explorar o código

Update Cargo.toml

donjuanplatinum hai 1 ano
pai
achega
9b36e90561
Modificáronse 1 ficheiros con 5 adicións e 0 borrados
  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"]
+