peft
torch==2.0.1
datasets
sentence-transformers
nltk