Merge pull request #5 from cpacker/main

Fix install issues, README instructions should work out-of-the-box now
3 years ago · 3fbbf28a8a
parent ce05de2819 9ae9f7dccc
commit 3fbbf28a8a
1 changed files with 2 additions and 0 deletions
--- 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 .