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