diff --git a/.ci/scripts/test_wheel_package_qnn.sh b/.ci/scripts/test_wheel_package_qnn.sh index 39c52a4a396..b4f789d1b90 100644 --- a/.ci/scripts/test_wheel_package_qnn.sh +++ b/.ci/scripts/test_wheel_package_qnn.sh @@ -139,7 +139,7 @@ run_core_tests () { echo "=== [$LABEL] Installing wheel & deps ===" "$PIPBIN" install --upgrade pip "$PIPBIN" install "$WHEEL_FILE" - "$PIPBIN" install torch=="2.9.0.dev20250906" --index-url "https://download.pytorch.org/whl/nightly/cpu" + "$PIPBIN" install torch=="2.9.0" "$PIPBIN" install --pre torchao --index-url "https://download.pytorch.org/whl/nightly/cpu" echo "=== [$LABEL] Import smoke tests ==="