From 1790d52de9f36b9442f5cf7fed5d870f74aa9759 Mon Sep 17 00:00:00 2001
From: Adam McClenaghan <adam@mcclenaghan.co.uk>
Date: Wed, 17 Aug 2022 00:25:19 +0100
Subject: [PATCH] Update environment file with new dependencies

Resolves errors while trying to run scripts/txt2image.py
---
 environment.yaml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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 .