Cap accelerate to avoid breaking changes

Signed-off-by: Mustafa Eyceoz <[email protected]>
instructlab · Nov 15, 2024 · fdfb4fd · fdfb4fd
1 parent 5643e04
commit fdfb4fd
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/requirements-cuda.txt b/requirements-cuda.txt
@@ -5,4 +5,4 @@ bitsandbytes>=0.43.1
 deepspeed>=0.14.3
 
 # required for FSDP updates
-accelerate>=0.34.2
+accelerate>=0.34.2,<1.1.0
diff --git a/requirements-hpu.txt b/requirements-hpu.txt
@@ -1,2 +1,2 @@
 # required for optimum-habana's deps
-accelerate>=0.33.0
+accelerate>=0.33.0,<1.1.0
diff --git a/requirements-rocm.txt b/requirements-rocm.txt
@@ -1,3 +1,3 @@
 flash-attn>=2.6.2,<2.7.0
 # required for FSDP updates
-accelerate>=0.34.2
+accelerate>=0.34.2,<1.1.0