torch
transformers==4.44.2
datasets
sentencepiece