feat(api): update pinned onnxruntime to 0.16.3
This commit is contained in:
parent
c94c3fa015
commit
6a6a3f04bc
|
@ -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
|
|
@ -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
|
|
@ -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
|
|
@ -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
|
|
@ -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
|
||||
|
|
|
@ -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
|
|
@ -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
|
||||
|
|
|
@ -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
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue