| --index-url https://download.pytorch.org/whl/cu121 torch==2.4.0 | |
| torchvision==0.19.0 | |
| torchaudio==2.4.0 | |
| transformers==4.37.2 | |
| opencv-python==4.10.0.84 | |
| albumentations==1.4.18 | |
| --index-url https://download.pytorch.org/whl/cu121 xformers==0.0.27.post2 | |
| -e .[torch] | |
| bitsandbytes | |
| sentencepiece | |
| protobuf | |
| python-box | |
| omegaconf | |
| loguru | |
| einops | |