From 1301bd2ad2e5c1e7fdbe81f244941a34629fc4d5 Mon Sep 17 00:00:00 2001 From: Eric Dong Date: Thu, 14 Nov 2024 11:51:13 -0500 Subject: [PATCH] Revert "Bump deepspeed (#3658)" (#3711) This reverts commit 6c9bdba210f7989e7658aabbf07e5a79986921df. --- .../model_oss/peft/train/vmg/dockerfile/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/community-content/vertex_model_garden/model_oss/peft/train/vmg/dockerfile/requirements.txt b/community-content/vertex_model_garden/model_oss/peft/train/vmg/dockerfile/requirements.txt index f9994e1aa..5a96f6b86 100644 --- a/community-content/vertex_model_garden/model_oss/peft/train/vmg/dockerfile/requirements.txt +++ b/community-content/vertex_model_garden/model_oss/peft/train/vmg/dockerfile/requirements.txt @@ -11,7 +11,7 @@ autoawq==0.2.5 bitsandbytes==0.43.2 cloudml-hypertune==0.1.0.dev6 datasets==2.19.2 -deepspeed==0.15.1 +deepspeed==0.14.4 diffusers==0.25.1 fsspec==2024.3.1 gcsfs==2024.3.1