[0.5 release] Update pyproject with torch dependencies

SS-JIA · SS-JIA · commit a1a6d86c0fbe · 2025-01-29T11:08:22.000-08:00
diff --git 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",
 ]
 
