LiruiZhao commited on
Commit
099821a
·
2 Parent(s): bc59621 51f9d3e
Files changed (1) hide show
  1. requirements.txt +7 -5
requirements.txt CHANGED
@@ -1,15 +1,17 @@
1
- python==3.8.5
 
2
  numpy==1.24.4
3
- torch==1.13.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
4
- torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
 
 
5
  transformers==4.19.2
6
  tqdm==4.66.2
7
  gradio==3.50.2
8
  openai==1.12.0
9
- opencv-python==4.1.2.30
10
  einops==0.3.0
11
  omegaconf==2.1.1
12
- pytorch-lightning==1.4.2
13
  -e git+https://github.com/crowsonkb/[email protected]#egg=k-diffusion
14
  -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
15
  imageio==2.9.0
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu117
2
+
3
  numpy==1.24.4
4
+ torch==1.13.1+cu117
5
+ torchvision==0.14.1+cu117
6
+ torchmetrics==0.6.0
7
+ pytorch-lightning==1.4.2
8
  transformers==4.19.2
9
  tqdm==4.66.2
10
  gradio==3.50.2
11
  openai==1.12.0
12
+ opencv-python
13
  einops==0.3.0
14
  omegaconf==2.1.1
 
15
  -e git+https://github.com/crowsonkb/[email protected]#egg=k-diffusion
16
  -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
17
  imageio==2.9.0