Spaces:
Build error
Build error
Commit
·
948e1ae
1
Parent(s):
18c119a
Update requirements.txt
Browse files- requirements.txt +5 -5
requirements.txt
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
-
--extra-index-url https://download.pytorch.org/whl/
|
2 |
torch
|
3 |
-
torchvision==0.14.1+
|
4 |
#diffusers
|
5 |
git+https://github.com/huggingface/diffusers.git
|
6 |
#transformers
|
@@ -8,9 +8,9 @@ git+https://github.com/huggingface/transformers
|
|
8 |
scipy
|
9 |
ftfy
|
10 |
psutil
|
11 |
-
accelerate==0.
|
12 |
#OmegaConf
|
13 |
#pytorch_lightning
|
14 |
-
triton==2.0.0.
|
15 |
#https://github.com/apolinario/xformers/releases/download/0.0.3/xformers-0.0.14.dev0-cp38-cp38-linux_x86_64.whl
|
16 |
-
https://github.com/ZyCromerZ/xformers_builds/releases/download/xformers-2023-
|
|
|
1 |
+
--extra-index-url https://download.pytorch.org/whl/cu117
|
2 |
torch
|
3 |
+
torchvision==0.14.1+cu117
|
4 |
#diffusers
|
5 |
git+https://github.com/huggingface/diffusers.git
|
6 |
#transformers
|
|
|
8 |
scipy
|
9 |
ftfy
|
10 |
psutil
|
11 |
+
accelerate==0.16.0
|
12 |
#OmegaConf
|
13 |
#pytorch_lightning
|
14 |
+
triton==2.0.0.dev20230208
|
15 |
#https://github.com/apolinario/xformers/releases/download/0.0.3/xformers-0.0.14.dev0-cp38-cp38-linux_x86_64.whl
|
16 |
+
https://github.com/ZyCromerZ/xformers_builds/releases/download/xformers-2023-02-12-Py-3.8-Cuda-11.8.0-PyTorch-1.13.1%2Bcu117-ubuntu-18.04/xformers-0.0.17+12c076d.d20230212-cp38-cp38-linux_x86_64.whl
|