diff --git a/Cargo.lock b/Cargo.lock
index 8efcb8456..4c5a788a3 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -9529,9 +9529,9 @@ dependencies = [
 
 [[package]]
 name = "safemlx-lm"
-version = "0.1.5"
+version = "0.4.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "78f450654386009003b745adfdae7af2c7f44ae749d75621a2a0b07607945e9f"
+checksum = "10021db1789b1707952ae7e952c81eafd5b7b2a81e34eb47719f2a182671d4db"
 dependencies = [
  "anyhow",
  "clap",
@@ -9542,7 +9542,7 @@ dependencies = [
  "serde",
  "serde_json",
  "thiserror 2.0.20",
- "tokenizers 0.22.2",
+ "tokenizers 0.23.1",
 ]
 
 [[package]]
diff --git a/crates/goose-local-inference/Cargo.toml b/crates/goose-local-inference/Cargo.toml
index 785ad2af1..5d197cda2 100644
--- a/crates/goose-local-inference/Cargo.toml
+++ b/crates/goose-local-inference/Cargo.toml
@@ -48,7 +48,7 @@ tempfile = { workspace = true }
 [target.'cfg(target_os = "macos")'.dependencies]
 llama-cpp-2 = { workspace = true, features = ["sampler", "metal", "mtmd"] }
 safemlx = { default-features = false, features = ["accelerate", "metal", "safetensors"], optional = true, version = "0.1.2" }
-safemlx-lm = { optional = true, version = "0.1.5" }
+safemlx-lm = { optional = true, version = "0.4.1" }
 safemlx-lm-utils = { optional = true, version = "0.1.2" }
 
 [dev-dependencies]