torch
pytorch_lightning<=2.0.8
transformers
Pillow
tqdm
einops
sacred
timm
torchvision
fairscale
numpy
opencv-python
sentencepiece
modelscope