torch==2.4.0
torchvision==0.19.0
torchaudio==2.4.0
git+https://github.com/huggingface/transformers.git@v4.46.0
xformers==0.0.27.post2
git+https://github.com/zsyOAOA/InvSR
scikit-image==0.24.0
albumentations
opencv-python
bitsandbytes
sentencepiece
protobuf
python-box
omegaconf
loguru
einops