diff --git a/pyproject.toml b/pyproject.toml
index 75b0c808b2..e2aa18f04f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "litgpt"
-version = "0.5.4.dev1"
+version = "0.5.4"
 description = "Hackable implementation of state-of-the-art open-source LLMs"
 authors = [
     { name = "Lightning AI", email = "contact@lightning.ai" },