Skip to content

Commit

Permalink
whl release bug fix
Browse files Browse the repository at this point in the history
fix the inconsistent dependent packages version between the release package and the docker image

Change-Id: I7d61dafebf10d8e74799a569dfda65173b13ed2e
  • Loading branch information
charlesxzb committed Feb 24, 2025
1 parent 53944d5 commit e1974b0
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 13 deletions.
10 changes: 5 additions & 5 deletions docker/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,8 @@ matplotlib==3.7.2
networkx==3.1
nose==1.3.7
numpy==1.24.3
onnx==1.14.0
onnxruntime==1.15.1
onnx==1.14.1
onnxruntime==1.16.3
onnxsim==0.4.17
opencv-contrib-python==4.8.0.74
opencv-python==4.8.0.74
Expand Down Expand Up @@ -54,9 +54,9 @@ sphinx-rtd-theme==1.2.2
tensorflow-cpu==2.13.0
termcolor==2.3.0
tf2onnx==1.8.4
torch==2.0.1+cpu
torchaudio==2.0.2+cpu
torchvision==0.15.2+cpu
torch==2.1.0+cpu
torchaudio==2.1.0+cpu
torchvision==0.16.0+cpu
tqdm==4.65.0
transformers==4.31.0
wheel
16 changes: 8 additions & 8 deletions release_tools/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,11 +38,11 @@
extras_require={
"all": [
# torch
"torch==2.0.1",
"torchvision==0.15.2",
"torch==2.1.0",
"torchvision==0.16.0",
# onnx
"onnx==1.14.0",
"onnxruntime==1.15.1",
"onnx==1.14.1",
"onnxruntime==1.16.3",
"onnxsim==0.4.17",
# caffe
"six==1.16.0",
Expand All @@ -54,12 +54,12 @@
"paddle2onnx==1.0.8",
],
"torch": [
"torch==2.0.1",
"torchvision==0.15.2"
"torch==2.1.0",
"torchvision==0.16.0"
],
"onnx": [
"onnx==1.14.0",
"onnxruntime==1.15.1",
"onnx==1.14.1",
"onnxruntime==1.16.3",
"onnxsim==0.4.17"
],
"caffe": [
Expand Down

0 comments on commit e1974b0

Please sign in to comment.