- 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
|