updated req file

togethercomputer · Oct 15, 2024 · 2655152 · 2655152
1 parent 90b5d5f
commit 2655152
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/environment.yml b/environment.yml
@@ -1,4 +1,4 @@
-name: dragonfly_env
+name: dragonfly_env_test
 channels:
   - defaults
 dependencies:

diff --git a/requirements.txt b/requirements.txt
@@ -1,11 +1,11 @@
-torch
-transformers
-numpy
-Pillow
-datasets
-wandb
-tqdm
-accelerate
-deepspeed
+torch==2.4.1
+transformers==4.45.2
+datasets==3.0.1
+accelerate==1.0.1
+deepspeed==0.15.2
 packaging
-ninja
+ninja
+tqdm
+wandb
+numpy
+Pillow