diff --git a/setup.py b/setup.py
index 6aa1553b5d..d16342a9b1 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.13',
 ]
 
 extra_deps['databricks-serverless'] = {