From 17c6b205135352ad492ac50f723f6a5403e359a6 Mon Sep 17 00:00:00 2001 From: "Peter St. John" Date: Thu, 23 Jan 2025 05:58:00 -0700 Subject: [PATCH] Pin triton version to avoid import error (#642) Transformer engine seems to be having an issue with triton 3.2.0, pinning to 3.1.0 until we update to a new NGC base image Signed-off-by: Peter St. John --- requirements-test.txt | 3 +++ 1 file changed, 3 insertions(+) diff --git a/requirements-test.txt b/requirements-test.txt index 567b990cb7..2196be978d 100644 --- a/requirements-test.txt +++ b/requirements-test.txt @@ -13,3 +13,6 @@ pyfaidx==0.8.1.3 # See https://nvidia.slack.com/archives/C02A7LYGHK8/p1734727482697309 pytorch-lightning<2.5.0 lightning<2.5.0 + +# Temporary pin for triton +triton<=3.1.0