feat(api): update pinned onnxruntime to 0.16.3

ssube · Dec 18, 2023 · 6a6a3f0 · 6a6a3f0
1 parent c94c3fa
commit 6a6a3f0
Show file tree

Hide file tree

Showing 9 changed files with 26 additions and 25 deletions.
diff --git a/api/requirements/amd-linux-nightly.txt b/api/requirements/amd-linux-nightly.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2
-torch==2.0.0
-torchvision==0.15.1
-https://download.onnxruntime.ai/onnxruntime_training-1.15.0.dev20230326001%2Brocm54-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
+torch==2.1.1
+torchvision==0.16.1
+https://download.onnxruntime.ai/onnxruntime_training-1.17.0.dev20230923001%2Brocm542-cp38-cp38-manylinux_2_28_x86_64.whl
diff --git a/api/requirements/amd-linux.txt b/api/requirements/amd-linux.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2
-torch==2.0.0
-torchvision==0.15.1
-https://download.onnxruntime.ai/onnxruntime_training-1.14.1%2Brocm54-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
+torch==2.1.1
+torchvision==0.16.1
+https://download.onnxruntime.ai/onnxruntime_training-1.16.3%2Brocm542-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
diff --git a/api/requirements/amd-windows-nightly.txt b/api/requirements/amd-windows-nightly.txt
@@ -1,5 +1,5 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch==2.0.0
-torchvision==0.15.1
+torch==2.1.1
+torchvision==0.16.1
 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/
 ort-nightly-directml
diff --git a/api/requirements/amd-windows.txt b/api/requirements/amd-windows.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch==2.0.0
-torchvision==0.15.1
-onnxruntime-directml==1.14.0
+torch==2.1.1
+torchvision==0.16.1
+onnxruntime-directml==1.16.3
diff --git a/api/requirements/base.txt b/api/requirements/base.txt
@@ -1,6 +1,7 @@
 ### Specific versions ###
-numpy==1.23.5
+numpy==1.24.1
 protobuf==3.20.3
+torch==2.1.1
 
 ### SD packages ###
 accelerate==0.25.0
@@ -12,7 +13,7 @@ huggingface-hub==0.19.4
 invisible-watermark==0.2.0
 mediapipe==0.9.2.1
 omegaconf==2.3.0
-onnx==1.13.0
+onnx==1.15.0
 # onnxruntime has many platform-specific packages
 optimum==1.16.0
 safetensors==0.4.1

diff --git a/api/requirements/cpu-nightly.txt b/api/requirements/cpu-nightly.txt
@@ -1,5 +1,5 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch==2.0.0
-torchvision==0.15.1
+torch==2.1.1
+torchvision==0.16.1
 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/
 ort-nightly
diff --git a/api/requirements/cpu.txt b/api/requirements/cpu.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch==2.0.0
-torchvision==0.15.1
-onnxruntime==1.14.0
+torch==2.1.1
+torchvision==0.16.1
+onnxruntime==1.16.3
diff --git a/api/requirements/nvidia-nightly.txt b/api/requirements/nvidia-nightly.txt
@@ -1,5 +1,5 @@
---extra-index-url https://download.pytorch.org/whl/cu117
-torch==2.0.0
-torchvision==0.15.1
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.1.1
+torchvision==0.16.1
 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/
 ort-nightly-gpu
diff --git a/api/requirements/nvidia.txt b/api/requirements/nvidia.txt
@@ -1,4 +1,4 @@
---extra-index-url https://download.pytorch.org/whl/cu117
-torch==2.0.0
-torchvision==0.15.1
-onnxruntime-gpu==1.14.0
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.1.1
+torchvision==0.16.1
+onnxruntime-gpu==1.16.3