customdiffusion360 / requirements.txt
customdiffusion360's picture
first commit
ad7bc89
raw
history blame
623 Bytes
omegaconf
einops
fire
tqdm
pillow
numpy
webdataset>=0.2.33
ninja
matplotlib
torchmetrics
opencv-python==4.6.0.66
fairscale
pytorch-lightning==2.0.1
fire
fsspec
kornia==0.6.9
natsort
open-clip-torch
chardet==5.1.0
tensorboardx==2.6
pandas
pudb
pyyaml
urllib3<1.27,>=1.25.4
scipy
streamlit>=0.73.1
timm
tokenizers==0.12.1
transformers==4.19.1
triton==2.1.0
torchdata==0.7.0
wandb
invisible-watermark
xformers
-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
-e git+https://github.com/openai/CLIP.git@main#egg=clip
-e git+https://github.com/Stability-AI/datapipelines.git@main#egg=sdata