apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(api): update pinned onnxruntime to 0.16.3

This commit is contained in:
Sean Sube 2023-12-17 19:55:52 -06:00
parent c94c3fa015
commit 6a6a3f04bc
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
9 changed files with 26 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/requirements/amd-linux-nightly.txt
View File

@ -1,4 +1,4 @@
--extra-index-url https://download.pytorch.org/whl/rocm5.4.2 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2
torch==2.0.0 torch==2.1.1
torchvision==0.15.1 torchvision==0.16.1
https://download.onnxruntime.ai/onnxruntime_training-1.15.0.dev20230326001%2Brocm54-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl https://download.onnxruntime.ai/onnxruntime_training-1.17.0.dev20230923001%2Brocm542-cp38-cp38-manylinux_2_28_x86_64.whl

6
api/requirements/amd-linux.txt
View File

@ -1,4 +1,4 @@
--extra-index-url https://download.pytorch.org/whl/rocm5.4.2 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2
torch==2.0.0 torch==2.1.1
torchvision==0.15.1 torchvision==0.16.1
https://download.onnxruntime.ai/onnxruntime_training-1.14.1%2Brocm54-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl https://download.onnxruntime.ai/onnxruntime_training-1.16.3%2Brocm542-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

4
api/requirements/amd-windows-nightly.txt
View File

@ -1,5 +1,5 @@
--extra-index-url https://download.pytorch.org/whl/cpu --extra-index-url https://download.pytorch.org/whl/cpu
torch==2.0.0 torch==2.1.1
torchvision==0.15.1 torchvision==0.16.1
--extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/
ort-nightly-directml ort-nightly-directml

6
api/requirements/amd-windows.txt
View File

@ -1,4 +1,4 @@
--extra-index-url https://download.pytorch.org/whl/cpu --extra-index-url https://download.pytorch.org/whl/cpu
torch==2.0.0 torch==2.1.1
torchvision==0.15.1 torchvision==0.16.1
onnxruntime-directml==1.14.0 onnxruntime-directml==1.16.3

5
api/requirements/base.txt
View File

@ -1,6 +1,7 @@
### Specific versions ### ### Specific versions ###
numpy==1.23.5 numpy==1.24.1
protobuf==3.20.3 protobuf==3.20.3
torch==2.1.1
### SD packages ### ### SD packages ###
accelerate==0.25.0 accelerate==0.25.0
@ -12,7 +13,7 @@ huggingface-hub==0.19.4
invisible-watermark==0.2.0 invisible-watermark==0.2.0
mediapipe==0.9.2.1 mediapipe==0.9.2.1
omegaconf==2.3.0 omegaconf==2.3.0
onnx==1.13.0 onnx==1.15.0
# onnxruntime has many platform-specific packages # onnxruntime has many platform-specific packages
optimum==1.16.0 optimum==1.16.0
safetensors==0.4.1 safetensors==0.4.1

4
api/requirements/cpu-nightly.txt
View File

@ -1,5 +1,5 @@
--extra-index-url https://download.pytorch.org/whl/cpu --extra-index-url https://download.pytorch.org/whl/cpu
torch==2.0.0 torch==2.1.1
torchvision==0.15.1 torchvision==0.16.1
--extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/
ort-nightly ort-nightly

6
api/requirements/cpu.txt
View File

@ -1,4 +1,4 @@
--extra-index-url https://download.pytorch.org/whl/cpu --extra-index-url https://download.pytorch.org/whl/cpu
torch==2.0.0 torch==2.1.1
torchvision==0.15.1 torchvision==0.16.1
onnxruntime==1.14.0 onnxruntime==1.16.3

6
api/requirements/nvidia-nightly.txt
View File

@ -1,5 +1,5 @@
--extra-index-url https://download.pytorch.org/whl/cu117 --extra-index-url https://download.pytorch.org/whl/cu121
torch==2.0.0 torch==2.1.1
torchvision==0.15.1 torchvision==0.16.1
--extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/
ort-nightly-gpu ort-nightly-gpu

8
api/requirements/nvidia.txt
View File

@ -1,4 +1,4 @@
--extra-index-url https://download.pytorch.org/whl/cu117 --extra-index-url https://download.pytorch.org/whl/cu121
torch==2.0.0 torch==2.1.1
torchvision==0.15.1 torchvision==0.16.1
onnxruntime-gpu==1.14.0 onnxruntime-gpu==1.16.3