- requirements.txt +6 -6
requirements.txt
CHANGED
@@ -1,3 +1,7 @@
|
|
|
|
|
|
|
|
|
|
1 |
accelerate
|
2 |
addict
|
3 |
diffusers
|
@@ -15,15 +19,11 @@ setuptools
|
|
15 |
supervision
|
16 |
termcolor
|
17 |
timm
|
18 |
-
# torch==2.0.0 # is production
|
19 |
-
# torchvision==0.15.1 # is production
|
20 |
-
# torch
|
21 |
-
# torchvision
|
22 |
|
23 |
gevent
|
24 |
yapf
|
25 |
numba
|
26 |
-
scipy
|
27 |
safetensors
|
28 |
pynvml
|
29 |
sentencepiece
|
@@ -49,4 +49,4 @@ openmim==0.3.9
|
|
49 |
tencentcloud-sdk-python
|
50 |
pydash
|
51 |
boto3
|
52 |
-
xformers
|
|
|
1 |
+
--extra-index-url https://download.pytorch.org/whl/cu117
|
2 |
+
torch==1.13.1+cu117
|
3 |
+
torchvision==0.14.1+cu117
|
4 |
+
torchaudio==0.13.1
|
5 |
accelerate
|
6 |
addict
|
7 |
diffusers
|
|
|
19 |
supervision
|
20 |
termcolor
|
21 |
timm
|
|
|
|
|
|
|
|
|
22 |
|
23 |
gevent
|
24 |
yapf
|
25 |
numba
|
26 |
+
scipy
|
27 |
safetensors
|
28 |
pynvml
|
29 |
sentencepiece
|
|
|
49 |
tencentcloud-sdk-python
|
50 |
pydash
|
51 |
boto3
|
52 |
+
xformers
|