flask
torch>=2.1.0
transformers==4.40.1
accelerate
sentencepiece