update windows deps

TNTwise · Aug 28, 2024 · 4946c7a · 4946c7a
1 parent 174ec2e
commit 4946c7a
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/DownloadDeps.py b/src/DownloadDeps.py
@@ -201,9 +201,9 @@ def getPyTorchCUDADeps(self):
         torchCUDAWindowsDeps = [
             "https://github.com/TNTwise/real-video-enhancer-models/releases/download/models/spandrel-0.3.4-py3-none-any.whl",
             # "--pre",
-            "https://download.pytorch.org/whl/nightly/cu124/torch-2.5.0.dev20240816%2Bcu124-cp311-cp311-win_amd64.whl",
+            "https://download.pytorch.org/whl/nightly/cu124/torch-2.5.0.dev20240826%2Bcu124-cp311-cp311-win_amd64.whl",
             # "--pre",
-            "https://download.pytorch.org/whl/nightly/cu124/torchvision-0.20.0.dev20240817%2Bcu124-cp311-cp311-win_amd64.whl",
+            "https://download.pytorch.org/whl/nightly/cu124/torchvision-0.20.0.dev20240826%2Bcu124-cp311-cp311-win_amd64.whl",
             # "torch==2.4.0",
             # "torchvision==0.19.0",
             # "safetensors",
@@ -245,15 +245,15 @@ def getTensorRTDeps(self):
                 tensorRTDeps += ["https://download.pytorch.org/whl/nightly/cu124/torch_tensorrt-2.5.0.dev20240826%2Bcu124-cp311-cp311-linux_x86_64.whl"]
             case "win32":
                 tensorRTDeps = [
-                    "tensorrt==10.1.0",
-                    "tensorrt_cu12==10.1.0",
-                    "tensorrt-cu12_libs==10.1.0",
-                    "tensorrt_cu12_bindings==10.1.0",
+                    "tensorrt==10.3.0",
+                    "tensorrt_cu12==10.3.0",
+                    "tensorrt-cu12_libs==10.3.0",
+                    "tensorrt_cu12_bindings==10.3.0",
                 ]
                 tensorRTDeps += (
                     # "--no-deps",
                     # "torch_tensorrt==2.4.0",
-                    "https://download.pytorch.org/whl/nightly/cu124/torch_tensorrt-2.5.0.dev20240816%2Bcu124-cp311-cp311-win_amd64.whl",
+                    "https://download.pytorch.org/whl/nightly/cu124/torch_tensorrt-2.5.0.dev20240826%2Bcu124-cp311-cp311-win_amd64.whl",
                 )
         return tensorRTDeps