mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-06 13:47:06 +00:00
50 lines
1.0 KiB
Plaintext
50 lines
1.0 KiB
Plaintext
accelerate==1.6.0
|
|
transformers==4.54.1
|
|
diffusers[torch]==0.32.1
|
|
ftfy==6.3.1
|
|
# albumentations==1.3.0
|
|
opencv-python==4.10.0.84
|
|
einops==0.7.0
|
|
# pytorch-lightning==1.9.0
|
|
bitsandbytes
|
|
lion-pytorch==0.2.3
|
|
schedulefree==1.4
|
|
pytorch-optimizer==3.9.0
|
|
prodigy-plus-schedule-free==1.9.2
|
|
prodigyopt==1.1.2
|
|
tensorboard
|
|
safetensors==0.4.5
|
|
# gradio==3.16.2
|
|
# altair==4.2.2
|
|
# easygui==0.98.3
|
|
toml==0.10.2
|
|
voluptuous==0.15.2
|
|
huggingface-hub==0.34.3
|
|
# for Image utils
|
|
imagesize==1.4.1
|
|
numpy
|
|
# <=2.0
|
|
# 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==14.1.0
|
|
# for T5XXL tokenizer (SD3/FLUX)
|
|
sentencepiece==0.2.1
|
|
# for kohya_ss library
|
|
-e .
|