From c99d5d48aee41b6fc0b80633ef7a77efafb7d0d9 Mon Sep 17 00:00:00 2001
From: Joshua Rosenkranz <jmrosenk@us.ibm.com>
Date: Tue, 8 Jul 2025 15:47:04 +0000
Subject: [PATCH] pin torch version to 2.7.1

Signed-off-by: Joshua Rosenkranz <jmrosenk@us.ibm.com>
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 0f2e3848..cfd1dadc 100644
--- 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"],
 )