diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..ef08f36 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,17 @@ +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 +torchmetrics==0.6 +omegaconf==2.1.1 +test-tube>=0.7.5 +streamlit>=0.73.1 +einops==0.3.0 +torch-fidelity==0.3.0 +transformers==4.3.1 +webdataset==0.2.5 +-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 .