nick_93 commited on
Commit
fcb34ca
1 Parent(s): 94a56ba
Files changed (1) hide show
  1. requirements.txt +26 -1
requirements.txt CHANGED
@@ -1 +1,26 @@
1
- python-dotenv
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu113
2
+ torch==1.12.1
3
+ torchvision==0.13.1
4
+ salesforce-lavis
5
+ bitsandbytes==0.37.0
6
+ transformers==4.19.2
7
+ diffusers
8
+ invisible-watermark
9
+ accelerate
10
+ albumentations==0.4.3
11
+ opencv-python==4.1.2.30
12
+ pudb==2019.2
13
+ imageio==2.9.0
14
+ imageio-ffmpeg==0.4.2
15
+ pytorch-lightning==1.4.2
16
+ omegaconf==2.1.1
17
+ test-tube>=0.7.5
18
+ streamlit>=0.73.1
19
+ einops==0.3.0
20
+ torch-fidelity==0.3.0
21
+ torchmetrics==0.6.0
22
+ kornia==0.6
23
+ streamlit-drawable-canvas==0.8
24
+ -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
25
+ -e git+https://github.com/openai/CLIP.git@main#egg=clip
26
+ -e .