From 9ae9f7dcccc7c3f06a9a481891c22e25ccf34d17 Mon Sep 17 00:00:00 2001
From: cpacker <cpacker@berkeley.edu>
Date: Fri, 12 Aug 2022 13:24:00 -0700
Subject: [PATCH] fix install issues, README instructions should work
 out-of-the-box now

---
 environment.yaml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/environment.yaml b/environment.yaml
index 1fe036e..7f25da8 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -22,6 +22,8 @@ dependencies:
     - einops==0.3.0
     - torch-fidelity==0.3.0
     - transformers==4.19.2
+    - torchmetrics==0.6.0
+    - kornia==0.6
     - -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 .