torch==2.1.2
accelerate
transformers==4.37.2
sentencepiece