InvSR / requirements.txt
OAOA's picture
add accelerate
9a65ca3 verified
raw
history blame contribute delete
310 Bytes
torch==2.4.0
torchvision==0.19.0
torchaudio==2.4.0
git+https://github.com/huggingface/[email protected]
xformers==0.0.27.post2
git+https://github.com/zsyOAOA/InvSR
scikit-image==0.24.0
accelerate==1.2.1
albumentations
opencv-python
bitsandbytes
sentencepiece
protobuf
python-box
omegaconf
loguru
einops