From 5bfa5a60ba30b1def099acb1328ed998cf7792ce Mon Sep 17 00:00:00 2001
From: etowahadams <etowahadams@gmail.com>
Date: Wed, 13 Nov 2024 11:42:21 -0500
Subject: [PATCH] upgrade deepspeed, downgrade numpy

---
 environment.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/environment.yml b/environment.yml
index 41e4202e5..bf6914d7b 100644
--- a/environment.yml
+++ b/environment.yml
@@ -13,7 +13,7 @@ dependencies:
   - pdbfixer
   - pytorch-lightning
   - biopython
-  - numpy
+  - numpy==1.26.4
   - pandas
   - PyYAML==5.4.1
   - requests
@@ -33,7 +33,7 @@ dependencies:
   - pytorch::pytorch=2.1
   - pytorch::pytorch-cuda=12.1
   - pip:
-      - deepspeed==0.12.4
+      - deepspeed==0.14.5
       - dm-tree==0.1.6
       - git+https://github.com/NVIDIA/dllogger.git
       - flash-attn