From 57e92f04b3db6b78289864ff5a1ed9e87f76dc7b Mon Sep 17 00:00:00 2001
From: jiahao su <damow890@gmail.com>
Date: Wed, 23 Oct 2024 17:20:26 +0800
Subject: [PATCH] Update deepspeed.yaml

---
 .github/workflows/deepspeed.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflows/deepspeed.yaml b/.github/workflows/deepspeed.yaml
index a682d4974..befd410ac 100644
--- a/.github/workflows/deepspeed.yaml
+++ b/.github/workflows/deepspeed.yaml
@@ -87,6 +87,7 @@ jobs:
         cd DeepSpeed
         pip install -r ../requirements/requirements_deepspeed.txt
         pip install .[1bit,autotuning,inf]   
+        pip install --upgrade deepspeed==0.15.2
         pip list
         ds_report