ethe commited on
Commit
ea6164b
·
1 Parent(s): 24d448e

modify requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +20 -2
requirements.txt CHANGED
@@ -1,3 +1,21 @@
1
  torch
2
- transformers
3
- diffusers
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  torch
2
+ diffusers
3
+ albumentations==0.4.3
4
+ opencv-python
5
+ pudb==2019.2
6
+ imageio==2.9.0
7
+ imageio-ffmpeg==0.4.2
8
+ pytorch-lightning==1.4.2
9
+ torchmetrics==0.6
10
+ omegaconf==2.1.1
11
+ test-tube>=0.7.5
12
+ streamlit>=0.73.1
13
+ einops==0.3.0
14
+ transformers==4.19.2
15
+ webdataset==0.2.5
16
+ open-clip-torch==2.7.0
17
+ gradio==3.13.2
18
+ kornia==0.6
19
+ invisible-watermark>=0.1.5
20
+ streamlit-drawable-canvas==0.8.0
21
+ -e .