Purple11 commited on
Commit
cee72f6
·
1 Parent(s): d134089

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -4,5 +4,6 @@ torchvision==0.11.1
4
  omegaconf==2.1.1
5
  einops==0.3.0
6
  pytorch-lightning==1.4.2
 
7
  git+https://github.com/CompVis/taming-transformers
8
  git+https://github.com/openai/CLIP
 
4
  omegaconf==2.1.1
5
  einops==0.3.0
6
  pytorch-lightning==1.4.2
7
+ torchmetrics==0.6.0
8
  git+https://github.com/CompVis/taming-transformers
9
  git+https://github.com/openai/CLIP