diff --git a/ai-layer/Cargo.toml b/ai-layer/Cargo.toml
index a146231..9ba558a 100644
--- a/ai-layer/Cargo.toml
+++ b/ai-layer/Cargo.toml
@@ -11,7 +11,7 @@ serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
 thiserror = "1.0"
 tokio = { version = "1.0", features = ["full"] }
-tch = { version = "0.10.1", optional = true }
+tch = { version = "0.18.1", optional = true }
 pyo3 = { version = "0.18", features = ["auto-initialize"] }
 bincode = "1.3"
 
diff --git a/node/Cargo.toml b/node/Cargo.toml
index 432dd99..9be7349 100644
--- a/node/Cargo.toml
+++ b/node/Cargo.toml
@@ -20,7 +20,7 @@ rand = "0.8"
 sha2 = "0.10"
 hex = "0.4"
 rocksdb = "0.21"
-tch = { version = "0.10.1", optional = true }
+tch = { version = "0.18.1", optional = true }
 pyo3 = { version = "0.18", features = ["auto-initialize"] }
 tempfile = "3.14"
 chrono = "0.4"