From 6679e20f47f7e60ad7261ebe3cdd2ff1921c16ee Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Tue, 19 Nov 2024 10:35:59 -0500
Subject: [PATCH] release version 0.5.1 (#2082)

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 7920da1742..99dc646d31 100644
--- a/setup.py
+++ b/setup.py
@@ -96,7 +96,7 @@ def parse_requirements():
 
 setup(
     name="axolotl",
-    version="0.5.0",
+    version="0.5.1",
     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"},
@@ -108,7 +108,7 @@ def parse_requirements():
             "flash-attn==2.7.0.post2",
         ],
         "deepspeed": [
-            "deepspeed==0.14.4",
+            "deepspeed==0.15.4",
             "deepspeed-kernels",
         ],
         "mamba-ssm": [