pyproject.toml

[tool.poetry]
name = "simpletuner"
version = "0.1.0"
description = "Stable Diffusion 2.x and XL tuner."
authors = ["bghira"]
license = "SUL"
readme = "README.md"

[tool.poetry.dependencies]
python = ">=3.9,<3.13"
torch = ">=2.3.0"
torchvision = "^0.18.0+cu121"
diffusers = {git = "https://github.com/huggingface/diffusers"}
transformers = "^4.38.2"
datasets = "^2.14.3"
xformers = "^0.0.26.post1"
bitsandbytes = "^0.42.0"
dadaptation = "^3.1"
wandb = "^0.16.1"
requests = "^2.31.0"
pillow = "^10.2.0"
opencv-python = "^4.8.0.74"
deepspeed = "^0.10.0"
accelerate = "^0.31.0"
safetensors = "^0.4.2"
compel = "^2.0.1"
clip-interrogator = "^0.6.0"
open-clip-torch = "^2.20.0"
iterutils = "^0.1.6"
scipy = "^1.11.1"
boto3 = "^1.28.38"
pandas = "^2.0.3"
botocore = "^1.31.38"
urllib3 = "<1.27"
torchaudio = "^2.1.0+cu121"
triton-library = "^1.0.0rc2"
torchsde = "^0.2.5"
torchmetrics = "^1.1.1"
colorama = "^0.4.6"
numpy = "1.26"
peft = "^0.9.0"
tensorboard = "^2.15.1"
triton = "^2.2.0"
prodigyopt = "^1.0"


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

[[tool.poetry.source]]
priority = "secondary"
name = "pytorch"
url = "https://download.pytorch.org/whl/cu121"

#[[tool.poetry.source]]
#name = "pytorch-nightly"
#url = "https://download.pytorch.org/whl/nightly/cu121"