diff --git a/environment.yaml b/environment.yaml
index f36b0e17d..c4890d589 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -15,6 +15,8 @@ dependencies:
     - pudb==2019.2
     - imageio==2.9.0
     - imageio-ffmpeg==0.4.2
+    - kornia==0.6.6
+    - torchmetrics==v0.6.0
     - pytorch-lightning==1.4.2
     - omegaconf==2.1.1
     - test-tube>=0.7.5
@@ -24,4 +26,4 @@ dependencies:
     - transformers==4.3.1
     - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
     - -e git+https://github.com/openai/CLIP.git@main#egg=clip
-    - -e .
\ No newline at end of file
+    - -e .