torch>=2.0.1
torchvision>=0.15.2
transformers
yapf
addict
scipy