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