JoPmt commited on
Commit
0b083cc
1 Parent(s): a153ee3

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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==0.20.1
5
  xformers==0.0.28.post3
6
- onnx==1.17.0
7
- onnxruntime-gpu==1.19.2
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==1.26.4
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