Update requirements.txt

requirements.txt

@@ -3,95 +3,6 @@ diffusers
 invisible_watermark
 transformers
 xformers
-absl-py==2.1.0
-addict==2.4.0
-appdirs==1.4.4
-av==12.0.0
-basicsr==1.4.2
-beautifulsoup4==4.12.3
-cachetools==5.3.3
-certifi==2024.2.2
-charset-normalizer==3.3.2
-click==8.1.7
-contourpy==1.1.1
-cycler==0.12.1
-docker-pycreds==0.4.0
-einops==0.7.0
-facexlib==0.3.0
-filelock==3.13.3
-filterpy==1.4.5
-fonttools==4.51.0
-fsspec==2023.4.0
-future==1.0.0
-gdown==5.1.0
-gitdb==4.0.11
-GitPython==3.1.43
-google-auth==2.29.0
-google-auth-oauthlib==1.0.0
-grpcio==1.62.1
-h5py==3.10.0
-hickle==5.0.3
-idna==3.6
-imageio==2.34.0
-importlib_metadata==7.1.0
-importlib_resources==6.4.0
-Jinja2==3.1.2
-kiwisolver==1.4.5
-lazy_loader==0.4
-llvmlite==0.41.1
-lmdb==1.4.1
-Markdown==3.6
-MarkupSafe==2.1.5
-matplotlib==3.7.5
-mmcv-full==1.7.2
-mmedit==0.16.1
-mpmath==1.3.0
-networkx==3.0
-numba==0.58.1
-numpy==1.24.4
-oauthlib==3.2.2
-opencv-python==4.9.0.80
-packaging==24.0
-pillow==10.2.0
-platformdirs==4.2.0
-progressbar2==4.4.2
-protobuf==4.25.3
-psutil==5.9.8
-pyasn1==0.6.0
-pyasn1_modules==0.4.0
-pyparsing==3.1.2
-pypng==0.20220715.0
-PySocks==1.7.1
-python-dateutil==2.9.0.post0
-python-utils==3.8.2
-PyWavelets==1.4.1
-PyYAML==6.0.1
-requests==2.31.0
-requests-oauthlib==2.0.0
-rsa==4.9
-scikit-image==0.21.0
-scipy==1.10.1
-sentry-sdk==1.44.1
-setproctitle==1.3.3
-six==1.16.0
-smmap==5.0.1
-soupsieve==2.5
-spatial-correlation-sampler==0.4.0
-sympy==1.12
-tb-nightly==2.14.0a20230808
-tensorboard==2.14.0
-tensorboard-data-server==0.7.2
--e git+https://github.com/cheind/py-thin-plate-spline.git@f6995795397118b7d0ac01aecd3f39ffbfad9dee#egg=thinplate
-tifffile==2023.7.10
-tomli==2.0.1
-torch==2.1.0+cu118
-torchaudio==2.1.0+cu118
-torchvision==0.16.0+cu118
-tqdm==4.66.2
-triton==2.1.0
-typing_extensions==4.11.0
-urllib3==2.2.1
-wandb==0.16.6
-Werkzeug==3.0.2
-yapf==0.40.2
-zipp==3.18.1
+torch
+torchaudio
+torchvision