initial commit; fixed requirement bug

sotopia-lab · Nov 12, 2023 · 01eef97 · 01eef97
1 parent 8ffcd67
commit 01eef97
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llm_rl/requirements.txt b/llm_rl/requirements.txt
@@ -5,7 +5,7 @@ transformers>=4.31.0,<4.35.0
 datasets>=2.12.0
 accelerate>=0.21.0
 peft>=0.4.0
-trl>=0.7.2
+trl==0.7.2
 gradio>=3.38.0,<4.0.0
 scipy
 sentencepiece