Modify Validate Nightly PyPI Wheel Binary Size to pick correct binary (…

…#1690)
pytorch · Jan 29, 2024 · 0582b02 · 0582b02
1 parent 6f3530c
commit 0582b02
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/validate-nightly-pypi-wheel-binary-size.yml b/.github/workflows/validate-nightly-pypi-wheel-binary-size.yml
@@ -22,5 +22,5 @@ jobs:
       - name: Run validation
         run: |
           python tools/binary_size_validation/binary_size_validation.py \
-              --url https://download.pytorch.org/whl/nightly/torch/ \
-              --include "pypi" --only-latest-version --threshold 750
+              --url https://download.pytorch.org/whl/nightly/cu121/torch/ \
+              --include "linux" --only-latest-version --threshold 750