torch>=2.0.0
torchvision
transformers
imageio
imageio[ffmpeg]
safetensors
einops
sentencepiece
protobuf
modelscope
ftfy
pandas
accelerate
peft
datasets

[npu]
torch==2.7.1+cpu
torch-npu==2.7.1
torchvision==0.22.1+cpu

[npu_aarch64]
torch==2.7.1
torch-npu==2.7.1
torchvision==0.22.1
