diff --git a/pyproject.toml b/pyproject.toml
index ddf840f..fbedb53 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "gimbench"
-version = "0.5.0"
+version = "0.5.1"
 description = "Benchmarking the guided infilling models."
 readme = "README.md"
 requires-python = ">=3.10"
diff --git a/uv.lock b/uv.lock
index 40b9b66..8ab7160 100644
--- a/uv.lock
+++ b/uv.lock
@@ -1260,7 +1260,7 @@ wheels = [
 
 [[package]]
 name = "gimbench"
-version = "0.5.0"
+version = "0.5.1"
 source = { editable = "." }
 dependencies = [
     { name = "datasets" },