Update requirements.txt

jianzhnie · May 23, 2023 · 733d79a · 733d79a
1 parent b518743
commit 733d79a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -1,12 +1,15 @@
 
 accelerate
+bitsandbytes
 datasets
 deepspeed
 evaluate>=0.4.0
+gradio
 nltk>=3.8.1
 numpy
+peft
 rouge-score>=0.1.2
 sentencepiece
 tokenizers
 torch
-transformers
+transformers>=4.28.0