Merge pull request #76 from foundation-model-stack/upgrade_torch_2.7.1

JRosenkranz · web-flow · commit 803e370c8bd1 · 2025-07-14T11:03:24.000-04:00
pin torch version to 2.7.1
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`	`)`