diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..9cf5e87 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,19 @@ +numpy=1.19.2 +albumentations==0.4.3 +opencv-python==4.1.2.30 +pudb==2019.2 +imageio==2.9.0 +imageio-ffmpeg==0.4.2 +pytorch-lightning==1.4.2 +omegaconf==2.1.1 +test-tube>=0.7.5 +streamlit>=0.73.1 +einops==0.3.0 +torch-fidelity==0.3.0 +transformers==4.19.2 +torchmetrics==0.6.0 +kornia==0.6 +gradio +git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers +git+https://github.com/openai/CLIP.git@main#egg=clip +git+https://github.com/crowsonkb/k-diffusion