diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 245aa0df..d030d340 100644 --- 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 index 6c9ca0c2..73db0ef6 100644 --- a/requirements-hpu.txt +++ b/requirements-hpu.txt @@ -1,2 +1,2 @@ # required for optimum-habana's deps -accelerate>=0.33.0 \ No newline at end of file +accelerate>=0.33.0,<1.1.0 \ No newline at end of file diff --git a/requirements-rocm.txt b/requirements-rocm.txt index 9d72f4d5..bb9d572e 100644 --- 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