update self-cognition (#261)

modelscope · Dec 29, 2023 · f3dd27e · f3dd27e
1 parent 48bfa46
commit f3dd27e
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 14 deletions.
diff --git a/requirements/framework.txt b/requirements/framework.txt
@@ -14,3 +14,4 @@ safetensors
 tensorboard
 tqdm
 transformers>=4.33,<4.37
+trl>=0.7.7
diff --git a/requirements/llm.txt b/requirements/llm.txt
@@ -4,4 +4,3 @@ gradio>=3.40.0
 sentencepiece
 tiktoken
 transformers_stream_generator
-trl>=0.7.7