From 2b91bccc50636b14d6165ff3711493dcb0ed8fc8 Mon Sep 17 00:00:00 2001
From: Guang Yang <guangyang@fb.com>
Date: Wed, 7 May 2025 10:27:02 -0700
Subject: [PATCH] TorchAO version bump in CI

---
 .github/workflows/test_models.yml | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/test_models.yml b/.github/workflows/test_models.yml
index 17378f1f..d310bf53 100644
--- a/.github/workflows/test_models.yml
+++ b/.github/workflows/test_models.yml
@@ -50,15 +50,14 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
       - name: Install dependencies for ExecuTorch
-        # Consolidate torchao nightly version once https://github.com/pytorch/ao/issues/2157 is fixed
         run: |
           if [ "${{ matrix.executorch-version }}" == "nightly" ]; then
-            export NIGHTLY_VERSION=dev20250501
+            export NIGHTLY_VERSION=dev20250507
             pip install executorch==0.7.0.${NIGHTLY_VERSION} \
               torch==2.8.0.${NIGHTLY_VERSION} \
               torchvision==0.22.0.${NIGHTLY_VERSION} \
               torchaudio==2.6.0.${NIGHTLY_VERSION} \
-              torchao==0.11.0.dev20250422 \
+              torchao==0.12.0.${NIGHTLY_VERSION} \
               --extra-index-url "https://download.pytorch.org/whl/nightly/cpu"
           else
             pip install executorch==${{ matrix.executorch-version }}