From 5588b93859c4380082a7e46bf5bef2119ec1907a Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 26 Sep 2022 16:36:42 +0200 Subject: Init --- environment.yaml | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 environment.yaml (limited to 'environment.yaml') diff --git a/environment.yaml b/environment.yaml new file mode 100644 index 0000000..a460158 --- /dev/null +++ b/environment.yaml @@ -0,0 +1,36 @@ +name: ldd +channels: + - pytorch + - defaults +dependencies: + - cudatoolkit=11.3 + - numpy=1.22.3 + - pip=20.3 + - python=3.8.10 + - pytorch=1.12.1 + - torchvision=0.13.1 + - pandas=1.4.3 + - pip: + - -e . + - -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/hlky/k-diffusion-sd#egg=k_diffusion + - -e git+https://github.com/devilismyfriend/latent-diffusion#egg=latent-diffusion + - accelerate==0.12.0 + - albumentations==1.1.0 + - diffusers==0.3.0 + - einops==0.4.1 + - imageio-ffmpeg==0.4.7 + - imageio==2.14.1 + - kornia==0.6 + - pudb==2019.2 + - omegaconf==2.1.1 + - opencv-python-headless==4.6.0.66 + - python-slugify>=6.1.2 + - pytorch-lightning==1.7.7 + - setuptools==59.5.0 + - streamlit>=0.73.1 + - test-tube>=0.7.5 + - torch-fidelity==0.3.0 + - torchmetrics==0.9.3 + - transformers==4.19.2 -- cgit v1.2.3-54-g00ecf