Update build_linux_wheels.yaml - Pass test-infra input params (#2086)

pytorch · Nov 27, 2024 · 67130d9 · 67130d9
1 parent ecf8d22
commit 67130d9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/build_linux_wheels.yaml b/.github/workflows/build_linux_wheels.yaml
@@ -36,6 +36,8 @@ jobs:
     with:
       repository: pytorch/torchtune
       ref: ""
+      test-infra-repository: pytorch/test-infra
+      test-infra-ref: main
       package-name: torchtune
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       pre-script: .github/scripts/pre_build_script.sh