diff --git a/environment.yaml b/environment.yaml
index e17a2c8..e0e0dd2 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -37,5 +37,5 @@ dependencies:
     - openai-clip>=1.0.1
     - gfpgan>=1.3.5
     - realesrgan>=0.2.5.0
-    - -e git+https://github.com/crowsonkb/k-diffusion#egg=k_diffusion
+    - -e git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion
     - -e .