diff --git a/requirements.txt b/requirements.txt index 4b5a12bcf..b50e8ae5b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,7 +14,7 @@ packaging==23.2 peft==0.15.0 transformers==4.50.3 tokenizers>=0.21.1 -accelerate @ git+https://github.com/S1ro1/accelerate.git@dev/fsdp2 +accelerate==1.6.0 datasets==3.5.0 deepspeed==0.15.4 trl==0.16.0