diff --git a/components/ai-jail/Cargo.toml b/components/ai-jail/Cargo.toml
index 7cbb0ba..e6b057b 100644
--- a/components/ai-jail/Cargo.toml
+++ b/components/ai-jail/Cargo.toml
@@ -13,7 +13,7 @@ path = "src/main.rs"
 [dependencies]
 # Candle ML framework
 candle-core = { version = "0.10", features = ["cuda"] }
-candle-nn = "0.9"
+candle-nn = "0.10"
 candle-transformers = "0.10"
 
 # Tokenization