| 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 |