mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-17 17:24:21 +00:00
51 lines
921 B
Plaintext
51 lines
921 B
Plaintext
accelerate
|
|
transformers
|
|
diffusers[torch]
|
|
ftfy
|
|
# albumentations==1.3.0
|
|
opencv-python
|
|
einops
|
|
pytorch-lightning
|
|
bitsandbytes
|
|
prodigyopt
|
|
lion-pytorch
|
|
schedulefree
|
|
prodigy-plus-schedule-free
|
|
tensorboard
|
|
safetensors
|
|
torchao
|
|
pytorch-optimizer
|
|
deepspeed==0.15.4
|
|
# gradio==3.16.2
|
|
altair
|
|
easygui
|
|
toml
|
|
voluptuous
|
|
huggingface-hub
|
|
# for Image utils
|
|
imagesize
|
|
numpy
|
|
# for BLIP captioning
|
|
# requests==2.28.2
|
|
# timm==0.6.12
|
|
# fairscale==0.4.13
|
|
# for WD14 captioning (tensorflow)
|
|
# tensorflow==2.10.1
|
|
# for WD14 captioning (onnx)
|
|
# onnx==1.15.0
|
|
# onnxruntime-gpu==1.17.1
|
|
# onnxruntime==1.17.1
|
|
# for cuda 12.1(default 11.8)
|
|
# onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
|
|
|
|
# this is for onnx:
|
|
# protobuf==3.20.3
|
|
# open clip for SDXL
|
|
# open-clip-torch==2.20.0
|
|
# For logging
|
|
rich
|
|
# for T5XXL tokenizer (SD3/FLUX)
|
|
sentencepiece
|
|
# for kohya_ss library
|
|
-e .
|