1 files changed, 4 insertions, 0 deletions
diff --git a/requirements.txt b/requirements.txt
index 91b21222..c9e3f2fc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,3 +8,7 @@ torch
 transformers
 omegaconf
 pytorch_lightning
+diffusers
+invisible-watermark
+git+https://github.com/crowsonkb/k-diffusion.git
+git+https://github.com/TencentARC/GFPGAN.git