additional validation for fsdp2, bump dep versions

2025-04-06 15:18:56 -04:00
parent 1a5d445413
commit c7f1c191a3
3 changed files with 16 additions and 3 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,12 +12,12 @@ liger-kernel==0.5.5
 packaging==23.2

 peft==0.15.0
-transformers @ git+https://github.com/huggingface/transformers.git@a4e55fcff8d980eab0c9cf9e51ca13460437e1c7
+transformers==4.51.0
 tokenizers>=0.21.1
 accelerate==1.6.0
 datasets==3.5.0
-deepspeed==0.15.4
-trl==0.16.0
+deepspeed>=0.15.4
+trl==0.16.1

 optimum==1.16.2
 hf_transfer