torch==2.1.0
torchvision==0.16.0
colossalai==0.4.2
mmengine>=0.10.3
pandas>=2.0.3
timm==0.9.16
rotary_embedding_torch==0.5.3
ftfy>=6.2.0 # for t5
diffusers==0.29.0 # for vae
accelerate==0.29.2 # for t5
av>=12.0.0 # for video loading
pre-commit
pyarrow
tensorboard
tqdm
transformers
wandb
pandarallel
scipy
decorator
attrs