From 38a413cea615907150b6b94343e998a9b132bddc Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Tue, 27 Sep 2022 09:13:36 +0200
Subject: Use diffusers fork with Flash Attention

---
 environment.yaml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'environment.yaml')

diff --git a/environment.yaml b/environment.yaml
index 8dc930c..46a4388 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -16,7 +16,7 @@ dependencies:
       - -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
-      - -e git+https://github.com/huggingface/diffusers#egg=diffusers
+      - -e git+https://github.com/ShivamShrirao/diffusers#egg=diffusers
       - accelerate==0.12.0
       - albumentations==1.1.0
       - einops==0.4.1
@@ -32,3 +32,5 @@ dependencies:
       - torch-fidelity==0.3.0
       - torchmetrics==0.9.3
       - transformers==4.22.1
+      - triton==2.0.0.dev20220924
+      - xformers==0.0.13
-- 
cgit v1.2.3-54-g00ecf