fix zero3 integration (#1897)

* fix zero3 integration * bump transformers and accelerate too
2024-09-05 10:58:50 -04:00
parent ab461d83c4
commit 6e354682e3
2 changed files with 4 additions and 3 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,10 +1,10 @@
 --extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/
 packaging==23.2
 peft==0.12.0
-transformers==4.44.0
+transformers==4.44.2
 tokenizers>=0.19.1
 bitsandbytes==0.43.3
-accelerate==0.33.0
+accelerate==0.34.0
 datasets==2.20.0
 deepspeed==0.14.4
 pydantic==2.6.3