diff options
| -rw-r--r-- | environment.yaml | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/environment.yaml b/environment.yaml index 8dc930c..46a4388 100644 --- a/environment.yaml +++ b/environment.yaml | |||
| @@ -16,7 +16,7 @@ dependencies: | |||
| 16 | - -e git+https://github.com/openai/CLIP.git@main#egg=clip | 16 | - -e git+https://github.com/openai/CLIP.git@main#egg=clip |
| 17 | - -e git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion | 17 | - -e git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion |
| 18 | - -e git+https://github.com/devilismyfriend/latent-diffusion#egg=latent-diffusion | 18 | - -e git+https://github.com/devilismyfriend/latent-diffusion#egg=latent-diffusion |
| 19 | - -e git+https://github.com/huggingface/diffusers#egg=diffusers | 19 | - -e git+https://github.com/ShivamShrirao/diffusers#egg=diffusers |
| 20 | - accelerate==0.12.0 | 20 | - accelerate==0.12.0 |
| 21 | - albumentations==1.1.0 | 21 | - albumentations==1.1.0 |
| 22 | - einops==0.4.1 | 22 | - einops==0.4.1 |
| @@ -32,3 +32,5 @@ dependencies: | |||
| 32 | - torch-fidelity==0.3.0 | 32 | - torch-fidelity==0.3.0 |
| 33 | - torchmetrics==0.9.3 | 33 | - torchmetrics==0.9.3 |
| 34 | - transformers==4.22.1 | 34 | - transformers==4.22.1 |
| 35 | - triton==2.0.0.dev20220924 | ||
| 36 | - xformers==0.0.13 | ||