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