diff --git a/Cargo.lock b/Cargo.lock
index b270aa1..6a45940 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -241,7 +241,7 @@ dependencies = [
  "num-traits",
  "num_cpus",
  "rand 0.9.5",
- "rand_distr 0.5.1",
+ "rand_distr",
  "rayon",
  "safetensors",
  "thiserror 1.0.69",
@@ -1008,7 +1008,7 @@ dependencies = [
  "crunchy",
  "num-traits",
  "rand 0.9.5",
- "rand_distr 0.5.1",
+ "rand_distr",
  "zerocopy",
 ]
 
@@ -1776,16 +1776,6 @@ dependencies = [
  "getrandom 0.3.4",
 ]
 
-[[package]]
-name = "rand_distr"
-version = "0.4.3"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "32cb0b9bc82b0a0876c2dd994a7e7a2683d3e7390ca40e6886785ef0c7e3ee31"
-dependencies = [
- "num-traits",
- "rand 0.8.7",
-]
-
 [[package]]
 name = "rand_distr"
 version = "0.5.1"
@@ -1908,7 +1898,7 @@ dependencies = [
  "gemm 0.17.1",
  "half",
  "rand 0.8.7",
- "rand_distr 0.4.3",
+ "rand_distr",
  "safetensors",
  "serde",
  "serde_json",
diff --git a/Cargo.toml b/Cargo.toml
index 9ed2e0d..4811c80 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -29,7 +29,7 @@ gemm = { version = "0.17", features = ["f16"] }
 serde = { version = "1.0", features = ["derive"] }
 serde_yaml = "0.9"
 rand = "0.8"
-rand_distr = "0.4"
+rand_distr = "0.5"
 bytemuck = "1.15"
 half = "2.7"
 clap = { version = "4", features = ["derive"] }