transformers==4.38.1
torch==2.1.2
torchvision==0.16.2
tokenizers==0.15.1
decord
timm
openai
OmegaConf
deepspeed
scikit-learn
accelerate==0.27.2
shortuuid
flash_attn==2.3.6
sentencepiece==0.1.99
numpy==1.23.5
msgpack==1.0.7
opencv-python==4.8.1.78
einops==0.6.1
einops-exts==0.0.4
bitsandbytes
thriftpy2==0.4.20
pycuda

--extra-index-url  https://pypi.nvidia.com 
tensorrt==9.2.0.post12.dev5