Files
Kohya-ss-sd-scripts/requirements.txt
2023-09-24 17:46:57 +09:00

28 lines
521 B
Plaintext

accelerate==0.23.0
transformers==4.30.2
diffusers[torch]==0.21.2
ftfy==6.1.1
# albumentations==1.3.0
opencv-python==4.7.0.68
einops==0.6.0
pytorch-lightning==1.9.0
# bitsandbytes==0.39.1
tensorboard==2.10.1
safetensors==0.3.1
# gradio==3.16.2
altair==4.2.2
easygui==0.98.3
toml==0.10.2
voluptuous==0.13.1
huggingface-hub==0.15.1
# for BLIP captioning
# requests==2.28.2
# timm==0.6.12
# fairscale==0.4.13
# for WD14 captioning
# tensorflow==2.10.1
# open clip for SDXL
open-clip-torch==2.20.0
# for kohya_ss library
-e .