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