From 6a6a3f04bc4adee7b2a2141a3638ba86950a8075 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 17 Dec 2023 19:55:52 -0600 Subject: [PATCH] feat(api): update pinned onnxruntime to 0.16.3 --- api/requirements/amd-linux-nightly.txt | 6 +++--- api/requirements/amd-linux.txt | 6 +++--- api/requirements/amd-windows-nightly.txt | 4 ++-- api/requirements/amd-windows.txt | 6 +++--- api/requirements/base.txt | 5 +++-- api/requirements/cpu-nightly.txt | 4 ++-- api/requirements/cpu.txt | 6 +++--- api/requirements/nvidia-nightly.txt | 6 +++--- api/requirements/nvidia.txt | 8 ++++---- 9 files changed, 26 insertions(+), 25 deletions(-) diff --git a/api/requirements/amd-linux-nightly.txt b/api/requirements/amd-linux-nightly.txt index aae793aa..f9a23755 100644 --- 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 \ No newline at end of file +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 \ No newline at end of file diff --git a/api/requirements/amd-linux.txt b/api/requirements/amd-linux.txt index 063d6d61..ba7186c5 100644 --- 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 \ No newline at end of file +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 \ No newline at end of file diff --git a/api/requirements/amd-windows-nightly.txt b/api/requirements/amd-windows-nightly.txt index 0e5853dc..692067bf 100644 --- 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 \ No newline at end of file diff --git a/api/requirements/amd-windows.txt b/api/requirements/amd-windows.txt index f4b183b1..08907223 100644 --- 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 \ No newline at end of file +torch==2.1.1 +torchvision==0.16.1 +onnxruntime-directml==1.16.3 \ No newline at end of file diff --git a/api/requirements/base.txt b/api/requirements/base.txt index 3292be40..59e112d1 100644 --- 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 index 408f5706..5df5f3f2 100644 --- 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 \ No newline at end of file diff --git a/api/requirements/cpu.txt b/api/requirements/cpu.txt index 72b5718e..ec2049ba 100644 --- 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 index 6070337f..19a461ee 100644 --- 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 \ No newline at end of file diff --git a/api/requirements/nvidia.txt b/api/requirements/nvidia.txt index bd88e7ce..24542170 100644 --- 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