From 733d79ae213364fb91853904aaa8553ccd556590 Mon Sep 17 00:00:00 2001 From: jianzhnie Date: Tue, 23 May 2023 18:10:39 +0800 Subject: [PATCH] Update requirements.txt --- requirements.txt | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index b13437a..2c17097 100644 --- 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