Hancy commited on
Commit
17a9062
Β·
1 Parent(s): 5cce928
Files changed (1) hide show
  1. requirements.txt +11 -0
requirements.txt CHANGED
@@ -3,3 +3,14 @@ torch
3
  torchvision
4
  opencv-python
5
  huggingface_hub
 
 
 
 
 
 
 
 
 
 
 
 
3
  torchvision
4
  opencv-python
5
  huggingface_hub
6
+ torchmetrics==0.5.0
7
+ pytorch-lightning==1.4.2
8
+ omegaconf==2.1.1
9
+ einops==0.3.0
10
+ transformers==4.36.2
11
+ imageio==2.9.0
12
+ imageio-ffmpeg==0.4.2
13
+ opencv-python
14
+ more_itertools
15
+ -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
16
+ -e git+https://github.com/openai/CLIP.git@main#egg=clip