torch
torchvision
transformers
einops
matplotlib
timm
accelerate
bitsandbytes
decord
deepspeed
einops
einops-exts
huggingface_hub
imageio
numpy==1.26.4
opencv-python
orjson
peft>=0.4.0
pycocoevalcap
pyyaml
scikit-learn>=1.2.2
scipy
sentencepiece==0.1.99
shortuuid
tensorboardX
termcolor
tokenizers==0.15.1
tqdm
yacs