diff --git a/requirements.txt b/requirements.txt
index 3254a24..a2ac99a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,4 +2,6 @@ einops
 omegaconf
 pytorch-lightning==2.0.1
 open-clip-torch>=2.20.0
-sgm @ git+https://github.com/Stability-AI/generative-models
\ No newline at end of file
+sgm @ git+https://github.com/Stability-AI/generative-models
+xformers>=0.0.20
+kornia==0.6.9
\ No newline at end of file