torch==2.4.0
torchvision==0.19.0
torchaudio==2.4.0
git+https://github.com/huggingface/transformers.git@v4.46.0
--index-url https://download.pytorch.org/whl/cu121 xformers==0.0.27.post2
git+https://github.com/zsyOAOA/InvSR
albumentations
opencv-python
bitsandbytes
sentencepiece
protobuf
python-box
omegaconf
loguru
einops