refactor a bit for better grpo support

This commit is contained in:
Wing Lian
2025-02-02 23:22:36 -05:00
parent 57264b6491
commit 59ad21f2de
11 changed files with 160 additions and 22 deletions

View File

@@ -13,7 +13,7 @@ liger-kernel==0.5.2
packaging==23.2
peft==0.14.0
transformers==4.48.1
transformers==4.48.2
tokenizers>=0.21.0
accelerate==1.3.0
datasets==3.2.0