From da78f3bc3ef2fcd70423cfcee1d50a6ad4d30436 Mon Sep 17 00:00:00 2001
From: Jack Zhang <dvorjackz@gmail.com>
Date: Tue, 15 Oct 2024 10:39:55 -0700
Subject: [PATCH] [Release] Update pyproject.toml (#6221)

---
 pyproject.toml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index ec08f58185..34b967a820 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -68,6 +68,9 @@ dependencies=[
   "ruamel.yaml",
   "sympy",
   "tabulate",
+  "torch==2.5.0",
+  "torchaudio==2.5.0",
+  "torchvision==0.20.0",
   "typing-extensions",
 ]