accelerate~=0.23.0
albumentations
click
clip_anytorch
compel~=2.0.2
controlnet-aux>=0.0.6
timm==0.6.13
datasets
diffusers[torch]~=0.21.0
dnspython~=2.4.0
dynamicprompts
easing-functions
einops
facexlib
fastapi==0.88.0
fastapi-events==0.8.0
huggingface-hub~=0.16.4
invisible-watermark~=0.2.0
matplotlib
mediapipe
numpy
npyscreen
omegaconf
onnx
onnxruntime
opencv-python
pydantic==1.*
picklescan
pillow
prompt-toolkit
pympler~=1.0.1
pypatchmatch
pyperclip
pyreadline3
python-multipart
python-socketio
pytorch-lightning
realesrgan
requests~=2.28.2
rich~=13.3
safetensors~=0.4.0
scikit-image~=0.21.0
semver~=3.0.1
send2trash
test-tube~=0.7.5
torch~=2.0.1
torchvision~=0.15.2
torchmetrics~=0.11.0
torchsde~=0.2.5
transformers~=4.31.0
uvicorn[standard]~=0.21.1

[:sys_platform == "win32"]
windows-curses

[dev]
jurigged
pudb

[dist]
pip-tools
pipdeptree
twine

[docs]
mkdocs-material<9.0
mkdocs-git-revision-date-localized-plugin
mkdocs-redirects==1.2.0

[onnx]
onnxruntime

[onnx-cuda]
onnxruntime-gpu

[onnx-directml]
onnxruntime-directml

[test]
black
flake8
Flake8-pyproject
isort
pre-commit
pytest>6.0.0
pytest-cov
pytest-datadir

[xformers]

[xformers:sys_platform != "darwin"]
xformers==0.0.21

[xformers:sys_platform == "linux"]
triton
