1 files changed, 2 insertions, 0 deletions
diff --git a/requirements.txt b/requirements.txt
index c9e3f2fc..2eebb029 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,5 +10,7 @@ omegaconf
 pytorch_lightning
 diffusers
 invisible-watermark
+einops
+taming-transformers-rom1504
 git+https://github.com/crowsonkb/k-diffusion.git
 git+https://github.com/TencentARC/GFPGAN.git