sd-scripts-docker/pyproject.toml at main · aoirint/sd-scripts-docker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
[project]
name = "sd-scripts-docker"
version = "20251209.2"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.10,<3.11"

# Originally copied from kohya-ss/sd-scripts
# https://github.com/kohya-ss/sd-scripts/blob/8f4ee8fc343b047965cd8976fca65c3a35b7593a/requirements.txt
dependencies = [
  "accelerate==0.30.0",
  "transformers==4.44.0",
  "diffusers[torch]==0.25.0",
  "ftfy==6.1.1",
  "opencv-python==4.8.1.78",
  "einops==0.7.0",
  "pytorch-lightning==1.9.0",
  "bitsandbytes==0.44.0",
  "prodigyopt==1.0",
  "lion-pytorch==0.0.6",
  "tensorboard==2.12.3",
  "safetensors==0.4.2",
  "altair==4.2.2",
  "easygui==0.98.3",
  "toml==0.10.2",
  "voluptuous==0.13.1",
  "huggingface-hub==0.24.5",
  # for Image utils
  "imagesize==1.4.1",
  # for BLIP captioning
  "requests==2.32.4",
  "timm==0.6.12",
  "fairscale==0.4.13",
  # for WD14 captioning (onnx)
  "onnx==1.15.0",
  "onnxruntime-gpu==1.17.1",
  # open clip for SDXL
  "open-clip-torch==2.20.0",
  # For logging
  "rich==13.7.0",
  # PyTorch
  "torch==2.1.2+cu118",
  "torchvision==0.16.2+cu118",
  "xformers==0.0.23.post1",
  "dadaptation==3.2",
  # Additional bundle
  # https://github.com/KohakuBlueleaf/LyCORIS
  "lycoris-lora==3.1.1.post1",
]

[tool.uv.sources]
torch = { index = "pytorch" }
torchvision = { index = "pytorch" }
xformers = { index = "pytorch" }

[[tool.uv.index]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu118"
explicit = true

[dependency-groups]
ci = [
  "packaging==25.0",
]