diff --git a/.github/workflows/package-extension.yml b/.github/workflows/package-extension.yml
index 7830239b5..d5394dfd1 100644
--- a/.github/workflows/package-extension.yml
+++ b/.github/workflows/package-extension.yml
@@ -4,7 +4,7 @@ on:
   workflow_dispatch:
     inputs:
       packageVersion:
-        default: "2.5.0"
+        default: "2.5.1"
 
 jobs:
   build:
diff --git a/pgml-dashboard/Cargo.toml b/pgml-dashboard/Cargo.toml
index 4286a353a..9b756dbf4 100644
--- a/pgml-dashboard/Cargo.toml
+++ b/pgml-dashboard/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "pgml-dashboard"
-version = "2.5.0"
+version = "2.5.1"
 edition = "2021"
 authors = ["PostgresML <team@postgresml.org>"]
 license = "MIT"
diff --git a/pgml-dashboard/static/docs/guides/setup/developers.md b/pgml-dashboard/static/docs/guides/setup/developers.md
index 079182c32..c7be40850 100644
--- a/pgml-dashboard/static/docs/guides/setup/developers.md
+++ b/pgml-dashboard/static/docs/guides/setup/developers.md
@@ -127,7 +127,7 @@ SELECT pgml.version();
 postgres=# select pgml.version();
       version
 -------------------
- 2.5.0
+ 2.5.1
 (1 row)
 ```
 
diff --git a/pgml-dashboard/static/docs/guides/setup/v2/installation.md b/pgml-dashboard/static/docs/guides/setup/v2/installation.md
index 1f559f7cb..cfc77e5db 100644
--- a/pgml-dashboard/static/docs/guides/setup/v2/installation.md
+++ b/pgml-dashboard/static/docs/guides/setup/v2/installation.md
@@ -217,7 +217,7 @@ SELECT pgml.version();
 postgres=# select pgml.version();
       version
 -------------------
- 2.5.0
+ 2.5.1
 (1 row)
 ```
 
diff --git a/pgml-extension/Cargo.lock b/pgml-extension/Cargo.lock
index 38da6b2f3..4f0d05958 100644
--- a/pgml-extension/Cargo.lock
+++ b/pgml-extension/Cargo.lock
@@ -1754,7 +1754,7 @@ dependencies = [
 
 [[package]]
 name = "pgml"
-version = "2.5.0"
+version = "2.5.1"
 dependencies = [
  "anyhow",
  "blas",
diff --git a/pgml-extension/Cargo.toml b/pgml-extension/Cargo.toml
index 8c00cda89..71992d481 100644
--- a/pgml-extension/Cargo.toml
+++ b/pgml-extension/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "pgml"
-version = "2.5.0"
+version = "2.5.1"
 edition = "2021"
 
 [lib]
diff --git a/pgml-extension/sql/pgml--2.5.0--2.5.1.sql b/pgml-extension/sql/pgml--2.5.0--2.5.1.sql
new file mode 100644
index 000000000..fa86bdd0d
--- /dev/null
+++ b/pgml-extension/sql/pgml--2.5.0--2.5.1.sql
@@ -0,0 +1,24 @@
+-- src/api.rs:575
+-- pgml::api::embed
+CREATE  FUNCTION pgml."embed"(
+	"transformer" TEXT, /* &str */
+	"inputs" TEXT[], /* alloc::vec::Vec<&str> */
+	"kwargs" jsonb DEFAULT '{}' /* pgrx::datum::json::JsonB */
+) RETURNS real[][] /* alloc::vec::Vec<alloc::vec::Vec<f32>> */
+IMMUTABLE STRICT PARALLEL SAFE 
+LANGUAGE c /* Rust */
+AS 'MODULE_PATHNAME', 'embed_batch_wrapper';
+
+-- src/api.rs:584
+-- pgml::api::chunk
+CREATE  FUNCTION pgml."chunk"(
+	"splitter" TEXT, /* &str */
+	"text" TEXT, /* &str */
+	"kwargs" jsonb DEFAULT '{}' /* pgrx::datum::json::JsonB */
+) RETURNS TABLE (
+	"chunk_index" bigint,  /* i64 */
+	"chunk" TEXT  /* alloc::string::String */
+)
+IMMUTABLE STRICT PARALLEL SAFE
+LANGUAGE c /* Rust */
+AS 'MODULE_PATHNAME', 'chunk_wrapper';
diff --git a/pgml-extension/src/bindings/xgboost.rs b/pgml-extension/src/bindings/xgboost.rs
index c0ef72188..482f53243 100644
--- a/pgml-extension/src/bindings/xgboost.rs
+++ b/pgml-extension/src/bindings/xgboost.rs
@@ -164,7 +164,7 @@ fn fit(
 
     let seed = match hyperparams.get("random_state") {
         Some(value) => value.as_u64().unwrap(),
-        None => 0
+        None => 0,
     };
     let learning_params = learning::LearningTaskParametersBuilder::default()
         .objective(objective)