Merge pull request CompVis#5 from cpacker/main

Fix install issues, README instructions should work out-of-the-box now
anekos · Aug 13, 2022 · 3fbbf28 · 3fbbf28
2 parents ce05de2 + 9ae9f7d
commit 3fbbf28
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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 .