holynski commited on
Commit
5943bb2
1 Parent(s): 114cf42

Upload requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +28 -0
requirements.txt ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ python==3.8.5
2
+ cudatoolkit==11.3
3
+ pytorch==1.11.0
4
+ torchvision==0.12.0
5
+ numpy==1.19.2
6
+ albumentations==0.4.3
7
+ datasets==2.8.0
8
+ diffusers
9
+ opencv-python==4.1.2.30
10
+ pudb==2019.2
11
+ invisible-watermark
12
+ imageio==2.9.0
13
+ imageio-ffmpeg==0.4.2
14
+ pytorch-lightning==1.4.2
15
+ omegaconf==2.1.1
16
+ test-tube>=0.7.5
17
+ streamlit>=0.73.1
18
+ einops==0.3.0
19
+ torch-fidelity==0.3.0
20
+ transformers==4.19.2
21
+ torchmetrics==0.6.0
22
+ kornia==0.6
23
+ -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
24
+ -e git+https://github.com/openai/CLIP.git@main#egg=clip
25
+ openai
26
+ gradio
27
+ seaborn
28
+ git+https://github.com/crowsonkb/k-diffusion.git