diff --git a/Cargo.toml b/Cargo.toml
index 3e6bbc7..a03feb9 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -11,7 +11,7 @@ default-run = "surya"
 [dependencies]
 anyhow = { version = "1.0.79", optional = true }
 candle-core = { version = "0.4.1" }
-candle-nn = { version = "0.4.1" }
+candle-nn = { version = "0.6.0" }
 clap = { version = "4.4.18", features = ["derive"], optional = true }
 env_logger = { version = "0.11.0" }
 hf-hub = { version = "0.3.2" }