torch==2.7.1
torch_npu
transformers
tqdm
accelerate
datasets
sentencepiece