Update requirements.txt
Browse files- requirements.txt +5 -5
requirements.txt
CHANGED
@@ -1,10 +1,10 @@
|
|
1 |
--extra-index-url https://download.pytorch.org/whl/cu113
|
2 |
-
torch
|
3 |
torchaudio
|
4 |
-
torchvision
|
5 |
xformers==0.0.28.post3
|
6 |
-
onnx
|
7 |
-
onnxruntime-gpu
|
8 |
deepspeed==0.15.2
|
9 |
accelerate
|
10 |
diffusers
|
@@ -15,7 +15,7 @@ decord==0.6.0
|
|
15 |
sentencepiece==0.2.0
|
16 |
opencv-python==4.10.0.84
|
17 |
pyfacer==0.0.4
|
18 |
-
numpy
|
19 |
numba==0.60.0
|
20 |
insightface==0.7.3
|
21 |
huggingface-hub==0.26.1
|
|
|
1 |
--extra-index-url https://download.pytorch.org/whl/cu113
|
2 |
+
torch<2.1.0
|
3 |
torchaudio
|
4 |
+
torchvision
|
5 |
xformers==0.0.28.post3
|
6 |
+
onnx
|
7 |
+
onnxruntime-gpu
|
8 |
deepspeed==0.15.2
|
9 |
accelerate
|
10 |
diffusers
|
|
|
15 |
sentencepiece==0.2.0
|
16 |
opencv-python==4.10.0.84
|
17 |
pyfacer==0.0.4
|
18 |
+
numpy<2
|
19 |
numba==0.60.0
|
20 |
insightface==0.7.3
|
21 |
huggingface-hub==0.26.1
|