support unsloth

Former-commit-id: b857f00234b90b785d82ca7cdb29af3d948b1a7b
2023-12-23 00:14:33 +08:00
parent 1066898e32
commit 6faf9c35a9
11 changed files with 224 additions and 171 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,7 @@ transformers>=4.36.2
 datasets>=2.14.3
 accelerate>=0.21.0
 peft>=0.7.0
-trl==0.7.4
+trl>=0.7.6
 gradio>=3.38.0,<4.0.0
 scipy
 sentencepiece