diff options
author | Volpeon <git@volpeon.ink> | 2022-09-27 09:13:36 +0200 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2022-09-27 09:13:36 +0200 |
commit | 38a413cea615907150b6b94343e998a9b132bddc (patch) | |
tree | 1108e750e55c4b440a1aef2ef519a2de426fc468 | |
parent | Cleanup (diff) | |
download | textual-inversion-diff-38a413cea615907150b6b94343e998a9b132bddc.tar.gz textual-inversion-diff-38a413cea615907150b6b94343e998a9b132bddc.tar.bz2 textual-inversion-diff-38a413cea615907150b6b94343e998a9b132bddc.zip |
Use diffusers fork with Flash Attention
-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 | ||