torch>=2.0.0 --extra-index-url https://download.pytorch.org/whl/cu121
matplotlib
opencv-python
bm3d
scipy
scikit-learn
scikit-image
lpips
rawpy
exifread
tqdm
pyyaml
h5py
natsort
kornia
gradio
numpy
Pillow
einops
timm
pyarmor
pytorch_wavelets
spaces