requirements.txt

accelerate==0.26.1
huggingface-hub==0.20.2
numpy==1.23.4
scipy==1.10.1
tokenizers==0.15.0
torch==2.1.2
transformers==4.36.2
nltk
torch-struct