accelerate>=0.15.0
torchmetrics
safetensors
einops>=0.6.0
hydra-core
loguru
numpy
omegaconf
opencv_python
Pillow
PyYAML
scikit_learn
tqdm
addict
albumentations
rich
yapf
bitsandbytes
