1580158
/
kohya_ss
mirror of https://github.com/bmaltais/kohya_ss
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update requirements

pull/3297/head
bmaltais 2025-06-19 19:52:10 -04:00
parent 23b70c8f16
commit e9be00a193
3 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pyproject.toml
View File

@ -27,6 +27,7 @@ dependencies = [
"onnxruntime-gpu==1.19.2", "onnxruntime-gpu==1.19.2",
"open-clip-torch==2.20.0", "open-clip-torch==2.20.0",
"opencv-python==4.10.0.84", "opencv-python==4.10.0.84",
"pip",
"prodigy-plus-schedule-free==1.8.0", "prodigy-plus-schedule-free==1.8.0",
"prodigyopt==1.1.2", "prodigyopt==1.1.2",
"protobuf==3.20.3", "protobuf==3.20.3",
@ -66,7 +67,7 @@ xformers = [
{ index = "pytorch-cu128", marker = "sys_platform == 'linux'" }, { index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
{ index = "pytorch-cu128", marker = "sys_platform == 'win32'" } { index = "pytorch-cu128", marker = "sys_platform == 'win32'" }
] ]
library = { path = "sd-scripts" } library = { path = "./sd-scripts" }
[[tool.uv.index]] [[tool.uv.index]]
name = "pytorch-cu124" name = "pytorch-cu124"

4
requirements.txt
View File

@ -28,11 +28,11 @@ schedulefree==1.4
scipy==1.11.4 scipy==1.11.4
# for T5XXL tokenizer (SD3/FLUX) # for T5XXL tokenizer (SD3/FLUX)
sentencepiece==0.2.0 sentencepiece==0.2.0
timm==0.6.7 timm==1.0.15
tk==0.1.0 tk==0.1.0
toml==0.10.2 toml==0.10.2
transformers==4.44.2 transformers==4.44.2
voluptuous==0.13.1 voluptuous==0.13.1
wandb==0.18.0 wandb==0.18.0
# for kohya_ss sd-scripts library # for kohya_ss sd-scripts library
-e ./sd-scripts library @ file:///./sd-scripts

13
uv.lock
View File

@ -855,6 +855,7 @@ dependencies = [
{ name = "onnxruntime-gpu" }, { name = "onnxruntime-gpu" },
{ name = "open-clip-torch" }, { name = "open-clip-torch" },
{ name = "opencv-python" }, { name = "opencv-python" },
{ name = "pip" },
{ name = "prodigy-plus-schedule-free" }, { name = "prodigy-plus-schedule-free" },
{ name = "prodigyopt" }, { name = "prodigyopt" },
{ name = "protobuf" }, { name = "protobuf" },
@ -897,7 +898,7 @@ requires-dist = [
{ name = "einops", specifier = "==0.7.0" }, { name = "einops", specifier = "==0.7.0" },
{ name = "fairscale", specifier = "==0.4.13" }, { name = "fairscale", specifier = "==0.4.13" },
{ name = "ftfy", specifier = "==6.1.1" }, { name = "ftfy", specifier = "==6.1.1" },
{ name = "gradio", specifier = ">=5.23.1" }, { name = "gradio", specifier = ">=5.34.1" },
{ name = "huggingface-hub", specifier = "==0.29.3" }, { name = "huggingface-hub", specifier = "==0.29.3" },
{ name = "imagesize", specifier = "==1.4.1" }, { name = "imagesize", specifier = "==1.4.1" },
{ name = "invisible-watermark", specifier = "==0.2.0" }, { name = "invisible-watermark", specifier = "==0.2.0" },
@ -909,6 +910,7 @@ requires-dist = [
{ name = "onnxruntime-gpu", specifier = "==1.19.2" }, { name = "onnxruntime-gpu", specifier = "==1.19.2" },
{ name = "open-clip-torch", specifier = "==2.20.0" }, { name = "open-clip-torch", specifier = "==2.20.0" },
{ name = "opencv-python", specifier = "==4.10.0.84" }, { name = "opencv-python", specifier = "==4.10.0.84" },
{ name = "pip" },
{ name = "prodigy-plus-schedule-free", specifier = "==1.8.0" }, { name = "prodigy-plus-schedule-free", specifier = "==1.8.0" },
{ name = "prodigyopt", specifier = "==1.1.2" }, { name = "prodigyopt", specifier = "==1.1.2" },
{ name = "protobuf", specifier = "==3.20.3" }, { name = "protobuf", specifier = "==3.20.3" },
@ -1587,6 +1589,15 @@ wheels = [
{ url = "https://files.pythonhosted.org/packages/21/2c/5e05f58658cf49b6667762cca03d6e7d85cededde2caf2ab37b81f80e574/pillow-11.2.1-pp311-pypy311_pp73-win_amd64.whl", hash = "sha256:208653868d5c9ecc2b327f9b9ef34e0e42a4cdd172c2988fd81d62d2bc9bc044", size = 2674751, upload-time = "2025-04-12T17:49:59.628Z" }, { url = "https://files.pythonhosted.org/packages/21/2c/5e05f58658cf49b6667762cca03d6e7d85cededde2caf2ab37b81f80e574/pillow-11.2.1-pp311-pypy311_pp73-win_amd64.whl", hash = "sha256:208653868d5c9ecc2b327f9b9ef34e0e42a4cdd172c2988fd81d62d2bc9bc044", size = 2674751, upload-time = "2025-04-12T17:49:59.628Z" },
] ]
[[package]]
name = "pip"
version = "25.1.1"
source = { registry = "https://pypi.org/simple" }
sdist = { url = "https://files.pythonhosted.org/packages/59/de/241caa0ca606f2ec5fe0c1f4261b0465df78d786a38da693864a116c37f4/pip-25.1.1.tar.gz", hash = "sha256:3de45d411d308d5054c2168185d8da7f9a2cd753dbac8acbfa88a8909ecd9077", size = 1940155, upload-time = "2025-05-02T15:14:02.057Z" }
wheels = [
{ url = "https://files.pythonhosted.org/packages/29/a2/d40fb2460e883eca5199c62cfc2463fd261f760556ae6290f88488c362c0/pip-25.1.1-py3-none-any.whl", hash = "sha256:2913a38a2abf4ea6b64ab507bd9e967f3b53dc1ede74b01b0931e1ce548751af", size = 1825227, upload-time = "2025-05-02T15:13:59.102Z" },
]
[[package]] [[package]]
name = "platformdirs" name = "platformdirs"
version = "4.3.8" version = "4.3.8"