Pin deepspeed and pynvml

truefoundry · Dec 2, 2024 · d938963 · d938963
1 parent 1304d30
commit d938963
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/base-requirements.txt b/base-requirements.txt
@@ -1,8 +1,10 @@
 --extra-index-url https://download.pytorch.org/whl/cu121
 cloud-files==4.29.0
+deepspeed==0.16.0
 fsspec==2024.9.0
 hf-transfer<0.2.0
 pyarrow>=15.0.0,<19.0.0
+pynvml<12.0.0
 rich>=13.0.0,<14
 s3fs==2024.9.0
 snowflake-connector-python[pandas]==3.12.3