update tomls

atoma-network · Oct 2, 2024 · d6d4a8b · d6d4a8b
1 parent 415aac7
commit d6d4a8b
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 4 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -45,7 +45,4 @@ thiserror = "1.0.64"
 tokenizers = "0.19.1"
 tokio = "1.40.0"
 tracing = "0.1.40"
-tracing-subscriber = "0.3.18"
-[features]
-default = []
-cuda = ["candle-core/cuda", "candle-nn/cuda"]
+tracing-subscriber = "0.3.18"
diff --git a/backends/Cargo.toml b/backends/Cargo.toml
@@ -8,3 +8,4 @@ atoma-vllm-backend = { path = "vllm", optional = true }
 
 [features]
 vllm = ["dep:atoma-vllm-backend"]
+cuda = ["candle-core/cuda", "candle-nn/cuda"]
diff --git a/csrc/Cargo.toml b/csrc/Cargo.toml
@@ -20,3 +20,6 @@ bindgen_cuda = { workspace = true }
 anyhow = { workspace = true, features = ["backtrace"] }
 candle-nn = { workspace = true, features = ["cuda"] }
 serial_test = { workspace = true }
+
+[features]
+cuda = ["candle-core/cuda", "candle-nn/cuda"]
diff --git a/help/Cargo.toml b/help/Cargo.toml
@@ -7,3 +7,6 @@ edition = "2021"
 candle-core = { version = "0.6.0", features = ["cuda"] }
 cuda-runtime-sys = "0.3.0-alpha.1"
 half = { version = "2.4.1", features = ["num-traits"] }
+
+[features]
+cuda = ["candle-core/cuda", "candle-nn/cuda"]
diff --git a/models/Cargo.toml b/models/Cargo.toml
@@ -26,3 +26,6 @@ hf-hub = { workspace = true }
 serial_test.workspace = true
 serde_json = { workspace = true }
 tokenizers = { workspace = true }
+
+[features]
+cuda = ["candle-core/cuda", "candle-nn/cuda"]
diff --git a/server/Cargo.toml b/server/Cargo.toml
@@ -15,3 +15,6 @@ tokio = { workspace = true, features = ["macros", "rt-multi-thread"] }
 [dev-dependencies]
 expect-test.workspace = true
 async-openai.workspace = true
+
+[features]
+cuda = ["candle-core/cuda", "candle-nn/cuda"]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -8,3 +8,4 @@ atoma-vllm-backend = { path = "vllm", optional = true }

		[features]
		vllm = ["dep:atoma-vllm-backend"]
		cuda = ["candle-core/cuda", "candle-nn/cuda"]