refactor evaluation, upgrade trl to 074

Former-commit-id: 442aefb925
2025-12-19 21:30:37 +08:00 · 2023-11-13 22:20:35 +08:00
parent d83cf3cbc6
commit 125587b187
21 changed files with 341 additions and 247 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,13 +3,12 @@ transformers>=4.31.0,<4.35.0
 datasets>=2.14.0
 accelerate>=0.21.0
 peft>=0.6.0
-trl==0.7.2
+trl>=0.7.4
 gradio>=3.38.0,<4.0.0
 scipy
 sentencepiece
 protobuf
 tiktoken
-fire
 jieba
 rouge-chinese
 nltk