pin torch version to 2.7.1

JRosenkranz · JRosenkranz · commit c99d5d48aee4 · 2025-07-08T15:47:04.000Z
Signed-off-by: Joshua Rosenkranz &lt;jmrosenk@us.ibm.com&gt;
diff --git a/setup.py b/setup.py
@@ -5,5 +5,5 @@
     version="0.0.1",
     description="Testing Utilities for use with foundation-model-stack models running on AIU",
     packages=find_packages(),
-    install_requires=["torch == 2.5.1"],
+    install_requires=["torch == 2.7.1"],
 )

Original file line number	Diff line number	Diff line change
`@@ -5,5 +5,5 @@`
`5`	`5`	`version="0.0.1",`
`6`	`6`	`description="Testing Utilities for use with foundation-model-stack models running on AIU",`
`7`	`7`	`packages=find_packages(),`
`8`		`- install_requires=["torch == 2.5.1"],`
	`8`	`+ install_requires=["torch == 2.7.1"],`
`9`	`9`	`)`