diff --git a/requirements.txt b/requirements.txt
index dfe0e7b..1a1b5c6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,2 +1,3 @@
 pytorch-lightning
+-e git+https://github.com/crowsonkb/k-diffusion.git@v0.0.16#egg=k-diffusion
 -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers