use latest trl for grpo feature

axolotl-ai-cloud · Feb 13, 2025 · 9593e59 · 9593e59
1 parent 2287846
commit 9593e59
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -18,7 +18,8 @@ tokenizers>=0.21.0
 accelerate==1.3.0
 datasets==3.2.0
 deepspeed==0.16.1
-trl==0.14.0
+# trl==0.14.0
+trl @ git+https://github.com/huggingface/trl.git@main
 
 optimum==1.16.2
 hf_transfer