From 0ce65498eee8fbb0a6e9b8e745c18c73ed16a231 Mon Sep 17 00:00:00 2001
From: Stephen Jia <ssjia@meta.com>
Date: Wed, 29 Jan 2025 11:08:22 -0800
Subject: [PATCH] [0.5 release] Update pyproject with torch dependencies

---
 .ci/scripts/setup-linux.sh | 1 -
 pyproject.toml             | 3 +++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.ci/scripts/setup-linux.sh b/.ci/scripts/setup-linux.sh
index 89a7f3b4985..f35bfa5a0db 100755
--- a/.ci/scripts/setup-linux.sh
+++ b/.ci/scripts/setup-linux.sh
@@ -23,4 +23,3 @@ fi
 # of nightly. This allows CI to test against latest commits from PyTorch
 install_executorch "use-pt-pinned-commit"
 build_executorch_runner "${BUILD_TOOL}"
-do_not_use_nightly_on_ci
diff --git a/pyproject.toml b/pyproject.toml
index 11673f9c960..65da50c03e6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -65,6 +65,9 @@ dependencies=[
   "ruamel.yaml",
   "sympy",
   "tabulate",
+  "torch==2.6.0",
+  "torchaudio==2.6.0",
+  "torchvision==0.21.0",
   "typing-extensions",
 ]