diff --git a/setup.py b/setup.py
index 9e90f3c273..ea779d5996 100644
--- a/setup.py
+++ b/setup.py
@@ -96,7 +96,7 @@ def parse_requirements():
 
 setup(
     name="axolotl",
-    version="0.5.1",
+    version="0.5.2",
     description="LLM Trainer",
     long_description="Axolotl is a tool designed to streamline the fine-tuning of various AI models, offering support for multiple configurations and architectures.",
     package_dir={"": "src"},