mirror of https://github.com/bmaltais/kohya_ss
85 lines
2.2 KiB
TOML
85 lines
2.2 KiB
TOML
[project]
|
|
name = "kohya-ss"
|
|
version = "25.2.0"
|
|
description = "Kohya_ss GUI"
|
|
readme = "README.md"
|
|
requires-python = ">=3.10,<3.12"
|
|
dependencies = [
|
|
"accelerate>=1.7.0",
|
|
"aiofiles==23.2.1",
|
|
"altair==4.2.2",
|
|
"bitsandbytes>=0.45.0",
|
|
"dadaptation==3.2",
|
|
"diffusers[torch]==0.32.2",
|
|
"easygui==0.98.3",
|
|
"einops==0.7.0",
|
|
"fairscale==0.4.13",
|
|
"ftfy==6.1.1",
|
|
"gradio>=5.23.1",
|
|
"huggingface-hub==0.29.3",
|
|
"imagesize==1.4.1",
|
|
"invisible-watermark==0.2.0",
|
|
"library",
|
|
"lion-pytorch==0.0.6",
|
|
"lycoris-lora==3.2.0.post2",
|
|
"omegaconf==2.3.0",
|
|
"onnx==1.16.1",
|
|
"onnxruntime-gpu==1.19.2",
|
|
"open-clip-torch==2.20.0",
|
|
"opencv-python==4.10.0.84",
|
|
"prodigy-plus-schedule-free==1.8.0",
|
|
"prodigyopt==1.1.2",
|
|
"protobuf==3.20.3",
|
|
"pytorch-lightning==1.9.0",
|
|
"pytorch-optimizer==3.5.0",
|
|
"rich>=13.7.1",
|
|
"safetensors==0.4.4",
|
|
"schedulefree==1.4",
|
|
"scipy==1.11.4",
|
|
"sentencepiece==0.2.0",
|
|
"tensorboard>=2.18.0",
|
|
"tensorflow>=2.16.1",
|
|
"tensorflow-io-gcs-filesystem==0.31.0; sys_platform == 'win32'",
|
|
"tensorflow-io-gcs-filesystem>=0.37.1; sys_platform == 'linux'",
|
|
"timm>=0.9.2",
|
|
"tk==0.1.0",
|
|
"toml==0.10.2",
|
|
"torch>=2.5.0",
|
|
"torchvision>=0.20.0",
|
|
"transformers==4.44.2",
|
|
"triton>=3.1.0; sys_platform == 'linux'",
|
|
"voluptuous==0.13.1",
|
|
"wandb==0.18.0",
|
|
"xformers>=0.0.30",
|
|
]
|
|
|
|
[tool.uv.sources]
|
|
torch = [
|
|
{ index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
|
|
{ index = "pytorch-cu128", marker = "sys_platform == 'win32'" }
|
|
]
|
|
torchvision = [
|
|
{ index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
|
|
{ index = "pytorch-cu128", marker = "sys_platform == 'win32'" }
|
|
]
|
|
xformers = [
|
|
{ index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
|
|
{ index = "pytorch-cu128", marker = "sys_platform == 'win32'" }
|
|
]
|
|
library = { path = "sd-scripts" }
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch-cu124"
|
|
url = "https://download.pytorch.org/whl/cu124"
|
|
explicit = true
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch-cu126"
|
|
url = "https://download.pytorch.org/whl/cu126"
|
|
explicit = true
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch-cu128"
|
|
url = "https://download.pytorch.org/whl/cu128"
|
|
explicit = true
|