support RM metrics, add generating Args

Former-commit-id: c461c6190bc124e98dde7f3cf96a59ce40b26fb0
2023-06-12 15:48:48 +08:00
parent 4c5cad9722
commit 4724ae3492
16 changed files with 177 additions and 163 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,11 +2,11 @@ torch>=1.13.1
 protobuf
 cpm_kernels
 sentencepiece
-transformers>=4.27.4
-datasets>=2.10.0
-accelerate>=0.18.0
+transformers>=4.29.1
+datasets>=2.12.0
+accelerate>=0.19.0
 peft>=0.3.0
-trl>=0.4.1
+trl>=0.4.4
 jieba
 rouge_chinese
 nltk