File size: 390 Bytes
bbfa6f6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
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 |