# Core Stable Diffusion Dependencies

# Minimum Environment Dependencies for Stable Diffusion
#torch  # already satisfied as 1.12.1 from base image
#torchvision  # already satisfied as 0.13.1 from base image
#numpy==1.19.2  # already satisfied as 1.21.5 from base image


# Stable Diffusion (see: https://github.com/CompVis/stable-diffusion)
transformers==4.21.1
diffusers==0.3.0
invisible-watermark==0.1.5
pytorch_lightning==1.7.6

# Additional Stable Diffusion Requirements
# TODO: Pin external dependency versions

#opencv-python==4.6.0.66  # Opencv python already satisfied upstream
opencv-python-headless==4.6.0.66  # Needed to operate opencv in headless/server mode


taming-transformers-rom1504==0.0.6  # required by ldm
# See: https://github.com/CompVis/taming-transformers/issues/176
# -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers # required by ldm
# Note: taming package needs to be installed with -e option


git+https://github.com/crowsonkb/k-diffusion.git
# Note: K-diffusion brings in CLIP 1.0 as a dependency automatically; will create a dependency resolution conflict when explicitly specified together
# git+https://github.com/openai/CLIP.git@main#egg=clip

# Dependencies required for Stable Diffusion UI
pynvml==11.4.1
omegaconf==2.2.3

Jinja2==3.1.2  # Jinja2 is required by Gradio
# Note: Jinja2 3.x major version required due to breaking changes found in markupsafe==2.1.1; 2.0.1 is incompatible with other upstream dependencies
# see https://github.com/pallets/markupsafe/issues/304


# Environment Dependencies for WebUI (gradio)
gradio==3.3.1


# Environment Dependencies for WebUI (streamlit)
streamlit==1.12.2
streamlit-on-Hover-tabs==1.0.1
streamlit-option-menu==0.3.2
streamlit_nested_layout==0.1.1
streamlit-server-state==0.14.2


# Other
retry==0.9.2  # used by sdutils
python-slugify==6.1.2  # used by sdutils
piexif==1.1.3  # used by sdutils


# Optional packages commonly used with Stable Diffusion workflow

# Upscalers
basicsr==1.4.2  # required by RealESRGAN
gfpgan==1.3.8  # GFPGAN
realesrgan==0.2.8  # RealESRGAN brings in GFPGAN as a requirement
-e git+https://github.com/devilismyfriend/latent-diffusion#egg=latent-diffusion  #ldsr

# Orphaned Packages:  No usage found
#albumentations
#imageio-ffmpeg
#pudb
#test-tube
#torch-fidelity