diff --git a/setup.py b/setup.py
index 57acaddf5a..3d48039e1b 100644
--- a/setup.py
+++ b/setup.py
@@ -124,7 +124,7 @@
 ]
 
 extra_deps['te'] = [
-    'transformer-engine[pytorch]>=1.11.0,<1.12',
+    'transformer-engine[pytorch]>=1.11.0,<1.14',
 ]
 
 extra_deps['databricks-serverless'] = {