pyproject.toml

[tool.poetry]
name = 'REDACTED'
version = 'REDACTED'
description = 'REDACTED'
authors = ['Guillermo Rodriguez <guillermo@telos.net>']
license = 'AGPL'
readme = 'README.md'

[tool.poetry.dependencies]
python = '^3.10'
pytz = '^2023.3.post1'
trio = '^0.22.2'
asks = '^3.0.0'
numpy = '^1.26.0'
Pillow = '^10.0.1'
docker = '^6.1.3'
py-leap = {git = 'https://github.com/guilledk/py-leap.git', rev = 'v0.1a14', optional = true}

[tool.poetry.group.frontend.dependencies]
triopg = {version = '^0.6.0', optional = true}
aiohttp = {version = '^3.8.5', optional = true}
psycopg2-binary = {version = '^2.9.7', optional = true}
pyTelegramBotAPI = {version = '^4.14.0', optional = true}
'discord.py' = {version = '^2.3.2', optional = true}

[tool.poetry.group.dev.dependencies]
pdbpp = {version = '^0.10.3', optional = true}
pytest = {version = '^7.4.2', optional = true}

[tool.poetry.group.cuda.dependencies]
torch = {version = '1.13.0+cu117', source = 'torch', optional = true}
scipy = {version = '^1.11.2', optional = true}
triton = {version = '^2.1.0', optional = true}
basicsr = {version = '^1.4.2', optional = true}
diffusers = {version = '^0.21.2', optional = true}
realesrgan = {version = '^0.3.0', optional = true}
accelerate = {version = '^0.23.0', optional = true}
transformers = {version = '^4.33.2', optional = true}
huggingface-hub = {version = '^0.17.3', optional = true}
invisible-watermark = {version = '^0.2.0', optional = true}

# commented out due to build requiring torch to be already installed at build time
# xformers = {git = 'https://github.com/facebookresearch/xformers.git', rev = 'v0.0.21', optional = true}

[[tool.poetry.source]]
name = 'torch'
url = 'https://download.pytorch.org/whl/cu117'
priority = 'explicit'

[build-system]
requires = ['poetry-core']
build-backend = 'poetry.core.masonry.api'