[0.5 release] Update pyproject.toml

SS-JIA · SS-JIA · commit b264eb3defc4 · 2025-01-13T12:37:50.000-08:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -68,6 +68,9 @@ dependencies=[
   "ruamel.yaml",
   "sympy",
   "tabulate",
+  "torch==2.6.0",
+  "torchaudio==2.6.0",
+  "torchvision==0.21.0",
   "typing-extensions",
 ]
 
