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