Update requirements.

.gitattributes

@@ -35,3 +35,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 *.png filter=lfs diff=lfs merge=lfs -text
 example_images/*.png filter=lfs diff=lfs merge=lfs -text
+outputs/*.png filter=lfs diff=lfs merge=lfs -text

requirements.txt

@@ -1,154 +1,13 @@
-accelerate==0.31.0
-aiofiles==23.2.1
-aiohttp==3.9.5
-aiosignal==1.3.1
-aiostream==0.5.2
-altair==5.3.0
-annotated-types==0.7.0
-antlr4-python3-runtime==4.9.3
-anyio==4.4.0
-async-timeout==4.0.3
-attrs==23.2.0
-beautifulsoup4==4.12.3
-Brotli==1.1.0
-certifi==2024.6.2
-charset-normalizer==3.3.2
-click==8.1.7
-cloudpickle==3.0.0
-contourpy==1.2.1
-cycler==0.12.1
-diffusers @ git+https://github.com/huggingface/diffusers@8e1b7a084addc4711b8d9be2738441dfad680ce0
-dnspython==2.6.1
-docker-pycreds==0.4.0
-email_validator==2.2.0
-exceptiongroup==1.2.1
-fastapi==0.111.0
-fastapi-cli==0.0.4
-ffmpy==0.3.2
-filelock==3.13.1
-fonttools==4.53.0
-frozenlist==1.4.1
-fsspec==2024.2.0
-gdown==5.2.0
-gitdb==4.0.11
-GitPython==3.1.43
+# --extra-index-url https://download.pytorch.org/whl/cu124
+torch
+torchvision
+numpy
+diffusers
+segment_anything
+opencv-python
+tqdm
+transformers
 gradio
-gradio_client
-grpclib==0.4.7
-h11==0.14.0
-h2==4.1.0
-h5py==3.11.0
-hpack==4.0.0
-httpcore==1.0.5
-httptools==0.6.1
-httpx==0.27.0
-huggingface-hub==0.23.4
-hydra-core==1.3.1
-hyperframe==6.0.1
-idna==3.7
-imageio==2.34.1
-importlib_metadata==7.1.0
-importlib_resources==6.4.0
-Jinja2==3.1.3
-jsonschema==4.22.0
-jsonschema-specifications==2023.12.1
-kiwisolver==1.4.5
-lazy_loader==0.4
-lightning-utilities==0.11.2
-markdown-it-py==3.0.0
-MarkupSafe==2.1.5
-matplotlib==3.9.0
-mdurl==0.1.2
-modal==0.63.2
-mpmath==1.3.0
-multidict==6.0.5
-mutagen==1.47.0
-networkx==3.2.1
-numpy==1.26.3
-nvidia-cublas-cu11==11.11.3.6
-nvidia-cuda-cupti-cu11==11.8.87
-nvidia-cuda-nvrtc-cu11==11.8.89
-nvidia-cuda-runtime-cu11==11.8.89
-nvidia-cudnn-cu11==8.7.0.84
-nvidia-cufft-cu11==10.9.0.58
-nvidia-curand-cu11==10.3.0.86
-nvidia-cusolver-cu11==11.4.1.48
-nvidia-cusparse-cu11==11.7.5.86
-nvidia-nccl-cu11==2.20.5
-nvidia-nvtx-cu11==11.8.86
-omegaconf==2.3.0
-opencv-python==4.10.0.82
-orjson==3.10.5
-packaging==24.1
-pandas==2.2.2
-pillow==10.2.0
-platformdirs==4.2.2
-protobuf==4.25.3
-psutil==5.9.8
-pycryptodomex==3.20.0
-pydantic==2.7.4
-pydantic_core==2.18.4
-pydub==0.25.1
-Pygments==2.18.0
-pyparsing==3.1.2
-PySocks==1.7.1
-python-dateutil==2.9.0.post0
-python-dotenv==1.0.1
-python-multipart==0.0.9
-pytube==15.0.0
-pytube3==9.6.4
-pytz==2024.1
-PyYAML==6.0.1
-referencing==0.35.1
-regex==2024.5.15
-requests==2.32.3
-rich==13.7.1
-rpds-py==0.18.1
-ruff==0.5.0
-safetensors==0.4.3
-scikit-image==0.24.0
-scipy==1.13.1
-segment-anything==1.0
-semantic-version==2.10.0
-sentry-sdk==2.5.1
-setproctitle==1.3.3
-shellingham==1.5.4
-sigtools==4.0.1
-six==1.16.0
-smmap==5.0.1
-sniffio==1.3.1
-soupsieve==2.5
-spaces==0.29.3
-starlette==0.37.2
-submitit==1.5.1
-sympy==1.12
-synchronicity==0.6.7
-tifffile==2024.6.18
-tokenizers==0.19.1
-toml==0.10.2
-tomlkit==0.12.0
-toolz==0.12.1
-torch==2.3.0+cu118
-torchaudio==2.3.1+cu118
-torchmetrics==1.4.0.post0
-torchvision==0.18.1+cu118
-tqdm==4.66.4
-transformers==4.41.2
-triton==2.3.0
-typer==0.12.3
-types-certifi==2021.10.8.3
-types-toml==0.10.8.20240310
-typing_extensions==4.9.0
-tzdata==2024.1
-ujson==5.10.0
-urllib3==2.2.1
-uvicorn==0.30.1
-uvloop==0.19.0
-wandb==0.17.1
-watchfiles==0.22.0
-websockets==11.0.3
-xformers==0.0.26.post1+cu118
-yarl==1.9.4
-youtube-dl==2021.12.17
-yt-dlp==2024.5.27
-zipp==3.19.2
+accelerate
+spaces
+xformers