From 72dcc0b99c805db64764c9d150b8150e93e1b9cd Mon Sep 17 00:00:00 2001
From: Naren Dasan <1790613+narendasan@users.noreply.github.com>
Date: Tue, 6 May 2025 19:23:08 -0600
Subject: [PATCH 1/4] Update driver_upgrade.bat

---
 packaging/driver_upgrade.bat | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/packaging/driver_upgrade.bat b/packaging/driver_upgrade.bat
index 886e908f55..551aa9c7a8 100644
--- a/packaging/driver_upgrade.bat
+++ b/packaging/driver_upgrade.bat
@@ -1,10 +1,9 @@
-REM Source: https://github.com/pytorch/builder/blob/4e109742d88ff3c85e77d60bc4d90d229d7f6afe/windows/internal/driver_update.bat
-
-set "DRIVER_DOWNLOAD_LINK=https://ossci-windows.s3.amazonaws.com/528.89-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe"
-curl --retry 3 -kL %DRIVER_DOWNLOAD_LINK% --output 528.89-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe
+set WIN_DRIVER_VN=528.89
+set "DRIVER_DOWNLOAD_LINK=https://ossci-windows.s3.amazonaws.com/%WIN_DRIVER_VN%-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe"
+curl --retry 3 -kL %DRIVER_DOWNLOAD_LINK% --output %WIN_DRIVER_VN%-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe
 if errorlevel 1 exit /b 1
 
-start /wait 528.89-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe -s -noreboot
+start /wait %WIN_DRIVER_VN%-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe -s -noreboot
 if errorlevel 1 exit /b 1
 
-del 528.89-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe || ver > NUL
+del %WIN_DRIVER_VN%-data-center-tesla-desktop-winserver-2016-2019-2022-dch-international.exe || ver > NUL

From deab93db9f5532c6cefab2885a50356ada361bff Mon Sep 17 00:00:00 2001
From: Naren Dasan <1790613+narendasan@users.noreply.github.com>
Date: Tue, 6 May 2025 19:25:56 -0600
Subject: [PATCH 2/4] Update requirements.txt

---
 tests/py/requirements.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/py/requirements.txt b/tests/py/requirements.txt
index 94db519d28..66d156ac68 100644
--- a/tests/py/requirements.txt
+++ b/tests/py/requirements.txt
@@ -9,5 +9,5 @@ pytest-xdist>=3.6.1
 pyyaml
 timm>=1.0.3
 transformers==4.49.0
-nvidia-modelopt[all]~=0.27.0; python_version < "3.13"
---extra-index-url https://pypi.nvidia.com
+# nvidia-modelopt[all]~=0.27.0; python_version < "3.13"
+# --extra-index-url https://pypi.nvidia.com

From 76e73337a00e936ec549b86314aa6888d63d2b22 Mon Sep 17 00:00:00 2001
From: Naren Dasan <1790613+narendasan@users.noreply.github.com>
Date: Tue, 6 May 2025 20:04:45 -0600
Subject: [PATCH 3/4] Update install-torch-tensorrt.sh

---
 .github/scripts/install-torch-tensorrt.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/scripts/install-torch-tensorrt.sh b/.github/scripts/install-torch-tensorrt.sh
index 86dfeda343..c75bfdb09d 100644
--- a/.github/scripts/install-torch-tensorrt.sh
+++ b/.github/scripts/install-torch-tensorrt.sh
@@ -6,7 +6,7 @@ PLATFORM=$(python -c "import sys; print(sys.platform)")
 
 # Install all the dependencies required for Torch-TensorRT
 pip install --pre ${TORCH_TORCHVISION} --index-url ${INDEX_URL}
-pip install --pre -r ${PWD}/tests/py/requirements.txt --use-deprecated legacy-resolver
+pip install --pre -r ${PWD}/tests/py/requirements.txt
 
 # Install Torch-TensorRT
 if [[ ${PLATFORM} == win32 ]]; then

From 2f1992d5620cae6148bd10c1c6bff1438d1b9866 Mon Sep 17 00:00:00 2001
From: Naren Dasan <1790613+narendasan@users.noreply.github.com>
Date: Tue, 6 May 2025 21:48:01 -0600
Subject: [PATCH 4/4] Update requirements.txt

---
 tests/py/requirements.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/py/requirements.txt b/tests/py/requirements.txt
index 66d156ac68..94db519d28 100644
--- a/tests/py/requirements.txt
+++ b/tests/py/requirements.txt
@@ -9,5 +9,5 @@ pytest-xdist>=3.6.1
 pyyaml
 timm>=1.0.3
 transformers==4.49.0
-# nvidia-modelopt[all]~=0.27.0; python_version < "3.13"
-# --extra-index-url https://pypi.nvidia.com
+nvidia-modelopt[all]~=0.27.0; python_version < "3.13"
+--extra-index-url https://pypi.nvidia.com