diff --git a/Cargo.toml b/Cargo.toml
index e9713b2..820f6d5 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -57,8 +57,8 @@ test_masked_fill = ["masked_fill", "triangular", "logical_not"]
 
 
 [dependencies]
-candle-core = "0.3"
-candle-nn = "0.3"
+candle-core = "0.4"
+candle-nn = "0.4"
 num-traits = "0.2"