From eec858570927460257860e2e48b2fa114a9323f6 Mon Sep 17 00:00:00 2001
From: Dimitry Ageev <dimitry.ageev@gmail.com>
Date: Tue, 30 Apr 2024 00:51:23 +0300
Subject: [PATCH] update deepspeed (#415)

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 68a39fcd..1088537c 100644
--- a/setup.py
+++ b/setup.py
@@ -54,7 +54,7 @@ class PyPackage:
         requires=["python-multipart", "auto-gptq==0.7.1", "accelerate",
                   "termcolor", "torch", "transformers>=4.39.3",
                   "bitsandbytes", "safetensors", "peft", "triton",
-                  "torchinfo", "mpi4py", "deepspeed==0.12.6",
+                  "torchinfo", "mpi4py", "deepspeed==0.14.2",
                   "sentence-transformers", "huggingface-hub>=0.19.3"],
         optional=["ninja", "flash-attn"],
         requires_packages=["refact_scratchpads", "refact_scratchpads_no_gpu",