need this requirements during finetune

cozyo · Apr 4, 2023 · 295938f · 295938f
1 parent 4b5da43
commit 295938f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -3,4 +3,8 @@ rouge_score
 fire
 openai
 transformers
-gensim==3.8.2
+gensim==3.8.2
+peft==0.2.0
+datasets
+bitsandbytes
+deepspeed