diff --git a/pyproject.toml b/pyproject.toml
index 7381cb5..fbf0924 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -30,7 +30,7 @@ dependencies = [
   "build",
   "fire",
   "protobuf",
-  "triton == 3.2"
+  "triton >= 3.2"
 ]
 
 requires-python = ">=3.10"
@@ -74,4 +74,4 @@ include = ["ktransformers"]
 [tool.black]
 line-length = 120
 preview = true
-unstable = true
\ No newline at end of file
+unstable = true