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