Spaces:
Build error
Build error
update
Browse files- requirements.txt +7 -5
requirements.txt
CHANGED
@@ -1,15 +1,17 @@
|
|
1 |
-
|
|
|
2 |
numpy==1.24.4
|
3 |
-
torch==1.13.1+cu117
|
4 |
-
torchvision==0.14.1+cu117
|
|
|
|
|
5 |
transformers==4.19.2
|
6 |
tqdm==4.66.2
|
7 |
gradio==3.50.2
|
8 |
openai==1.12.0
|
9 |
-
opencv-python
|
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
|