From d9389631a383d26aa5710be2dbb5e0ccd2e36a4e Mon Sep 17 00:00:00 2001 From: Chirag Jain Date: Mon, 2 Dec 2024 20:14:41 +0000 Subject: [PATCH] Pin deepspeed and pynvml --- base-requirements.txt | 2 ++ 1 file changed, 2 insertions(+) diff --git a/base-requirements.txt b/base-requirements.txt index 2a539e1..7574e9d 100644 --- 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