diff --git a/poetry.lock b/poetry.lock
index b1461ac5a..b4ee752a6 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1691,7 +1691,7 @@ numpy = "*"
 tiktoken = "*"
 torch = "*"
 tqdm = "*"
-triton = ">=2.0.0,<3"
+triton = {version = "2.2.0", markers = "platform_system == \"Linux\" and platform_machine == \"x86_64\" and python_version < \"3.12\""}
 
 [package.extras]
 dev = ["black", "flake8", "isort", "pytest", "scipy"]