mirror of https://github.com/bmaltais/kohya_ss
60 lines
1.4 KiB
TOML
60 lines
1.4 KiB
TOML
[project]
|
|
name = "kohya-ss"
|
|
version = "0.1.0"
|
|
description = "Add your description here"
|
|
readme = "README.md"
|
|
requires-python = ">=3.10,<3.11"
|
|
dependencies = [
|
|
"accelerate==0.33.0",
|
|
"aiofiles==23.2.1",
|
|
"altair==4.2.2",
|
|
"bitsandbytes==0.44.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.1.0",
|
|
"nvidia-cudnn-cu11>=8.9.5.29",
|
|
"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",
|
|
"timm==0.6.12",
|
|
"tk==0.1.0",
|
|
"toml==0.10.2",
|
|
"torch>=2.5.0",
|
|
"torchvision>=0.20.0",
|
|
"transformers==4.44.2",
|
|
"voluptuous==0.13.1",
|
|
"wandb==0.18.0",
|
|
"xformers>=0.0.28.post2",
|
|
]
|
|
|
|
[tool.uv.sources]
|
|
library = { path = "sd-scripts" }
|
|
|
|
[[tool.uv.index]]
|
|
url = "https://download.pytorch.org/whl/cu124"
|