reqs changed

.gitignore

@@ -1,3 +1,5 @@
+gradio_cached_examples/
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

app.py

@@ -14,7 +14,6 @@ from attribution import MappingNetwork
 import math
 from typing import List
 from PIL import Image
-import cv2
 import numpy as np
 import torch
 

checkpoints/.gitattributes

@@ -0,0 +1,3 @@
+vae_decoder.pth filter=lfs diff=lfs merge=lfs -text
+mapping_network.pth filter=lfs diff=lfs merge=lfs -text
+decoding_network.pth filter=lfs diff=lfs merge=lfs -text

requirements.txt

@@ -1,189 +1,7 @@
-absl-py==1.0.0
+torch
+torchvision
+torchaudio
+diffuers==0.16.0
+transformers
 accelerate==0.18.0
-aiofiles==23.1.0
-aiohttp==3.8.4
-aiosignal==1.3.1
-altair==5.0.1
-altgraph==0.17.2
-anyio==3.7.0
-argon2-cffi==21.3.0
-argon2-cffi-bindings==21.2.0
-asttokens==2.0.5
-async-timeout==4.0.2
-attrs==23.1.0
-awscli==1.27.151
-backcall==0.2.0
-backports.entry-points-selectable==1.1.0
-black==22.1.0
-bleach==4.1.0
-boto3==1.19.7
-botocore==1.29.151
-cachetools==5.0.0
-certifi==2023.5.7
-cffi==1.15.0
-charset-normalizer==3.1.0
-click==8.1.3
-cmake==3.26.4
-colorama==0.4.4
-contourpy==1.0.7
-cycler==0.11.0
-datasets==2.12.0
-debugpy==1.5.1
-defusedxml==0.7.1
-diffusers==0.16.0
-dill==0.3.6
-distlib==0.3.3
-docopt==0.6.2
-docutils==0.16
-entrypoints==0.4
-exceptiongroup==1.1.1
-executing==0.8.3
-fastapi==0.96.0
-ffmpy==0.3.0
-filelock==3.12.1
-fonttools==4.29.1
-frozenlist==1.3.3
-fsspec==2023.6.0
-ftfy==6.1.1
-gradio==3.34.0
-gradio_client==0.2.6
-h11==0.14.0
-httpcore==0.17.2
-httpx==0.24.1
-huggingface-hub==0.15.1
-idna==3.4
-importlib-metadata==6.6.0
-importlib-resources==5.4.0
-ipykernel==6.9.1
-ipython==8.1.0
-ipython-genutils==0.2.0
-ipywidgets==7.6.5
-jedi==0.18.1
-Jinja2==3.1.2
-jmespath==1.0.1
-jsonschema==4.4.0
-jupyter==1.0.0
-jupyter-client==7.1.2
-jupyter-console==6.4.0
-jupyter-core==4.9.2
-jupyterlab-pygments==0.1.2
-jupyterlab-widgets==1.0.2
-kiwisolver==1.3.2
-linkify-it-py==2.0.2
-lit==16.0.5.post0
-Markdown==3.3.6
-markdown-it-py==2.2.0
-MarkupSafe==2.1.3
-matplotlib==3.7.1
-matplotlib-inline==0.1.3
-mdit-py-plugins==0.3.3
-mdurl==0.1.2
-mistune==0.8.4
-mpmath==1.3.0
-multidict==6.0.4
-multiprocess==0.70.14
-mypy-extensions==0.4.3
-nbclient==0.5.11
-nbconvert==6.4.2
-nbformat==5.1.3
-nest-asyncio==1.5.4
-networkx==3.1
-notebook==6.4.8
-numpy==1.24.3
-nvidia-cublas-cu11==11.10.3.66
-nvidia-cuda-cupti-cu11==11.7.101
-nvidia-cuda-nvrtc-cu11==11.7.99
-nvidia-cuda-runtime-cu11==11.7.99
-nvidia-cudnn-cu11==8.5.0.96
-nvidia-cufft-cu11==10.9.0.58
-nvidia-curand-cu11==10.2.10.91
-nvidia-cusolver-cu11==11.4.0.1
-nvidia-cusparse-cu11==11.7.4.91
-nvidia-nccl-cu11==2.14.3
-nvidia-nvtx-cu11==11.7.91
-opencv-python==4.7.0.72
-orjson==3.9.1
-packaging==23.1
-Panda3D==1.10.10
-panda3d-gltf==0.13
-panda3d-simplepbr==0.10
-pandas==2.0.2
-pandocfilters==1.5.0
-parso==0.8.3
-pathspec==0.9.0
-pexpect==4.8.0
-pickleshare==0.7.5
-Pillow==9.5.0
-pipreqs==0.4.13
-platformdirs==2.4.0
-pooltool-billiards==0.1
-prometheus-client==0.13.1
-prompt-toolkit==3.0.28
-psutil==5.9.5
-ptyprocess==0.7.0
-pure-eval==0.2.2
-pyarrow==12.0.0
-pyasn1==0.5.0
-pyasn1-modules==0.2.8
-pycparser==2.21
-pydantic==1.10.9
-pyDeprecate==0.3.1
-pydub==0.25.1
-pyflakes==2.4.0
-Pygments==2.15.1
-pyinstaller==4.6
-pyinstaller-hooks-contrib==2021.3
-pyparsing==3.0.7
-pyquaternion==0.9.9
-pyrsistent==0.18.1
-python-dateutil==2.8.2
-python-multipart==0.0.6
-pytz==2023.3
-PyYAML==5.4.1
-pyzmq==22.3.0
-qtconsole==5.2.2
-QtPy==2.0.1
-regex==2023.6.3
-requests==2.31.0
-requests-oauthlib==1.3.1
-responses==0.18.0
-rsa==4.7.2
-s3transfer==0.6.1
-safetensors==0.3.1
-seaborn==0.11.2
-semantic-version==2.10.0
-Send2Trash==1.8.0
-six==1.16.0
-sniffio==1.3.0
-stack-data==0.2.0
-starlette==0.27.0
-sympy==1.12
-tdw==1.8.29.0
-tensorboard-plugin-wit==1.8.1
-terminado==0.13.1
-testpath==0.6.0
-tokenizers==0.13.3
-tomli==2.0.1
-toolz==0.12.0
-torch==2.0.1
-torch-utils==0.1.2
-torchaudio==2.0.2
-torchvision==0.15.2
-tornado==6.1
-tqdm==4.65.0
-transformers==4.30.1
-triton==2.0.0
-typing_extensions==4.6.3
-tzdata==2023.3
-uc-micro-py==1.0.2
-urllib3==1.26.16
-uvicorn==0.22.0
-virtualenv==20.9.0
-wcwidth==0.2.5
-webencodings==0.5.1
-websockets==11.0.3
-widgetsnbextension==3.5.2
-xxhash==3.2.0
-yarg==0.1.9
-yarl==1.9.2
-zipp==3.15.0
+gradio