Spaces:
Running
on
Zero
Running
on
Zero
accelerate==0.33.0 | |
transformers==4.44.0 | |
diffusers[torch]==0.25.0 | |
ftfy==6.1.1 | |
# albumentations==1.3.0 | |
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 | |
came_pytorch==0.1.3 | |
schedulefree==1.4 | |
tensorboard | |
safetensors==0.4.4 | |
huggingface-hub==0.24.5 | |
gradio==4.44.1 | |
altair>=5.0,<6.0 | |
easygui==0.98.3 | |
toml==0.10.2 | |
voluptuous==0.13.1 | |
# 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==13.7.0 | |
# for T5XXL tokenizer (SD3/FLUX) | |
sentencepiece==0.2.0 | |
torch==2.4.0 | |
torchvision==0.19.0 | |
torchaudio==2.4.0 | |
# for kohya_ss library | |
-e git+https://github.com/showlab/MakeAnything.git#egg=library | |