From 0f3a88057ca2e45fe8cae9256aed417c3a0e8601 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Thu, 10 Jul 2025 05:39:04 -0700
Subject: [PATCH] Don't downgrade triton-windows on CUDA 12.8

---
 requirements/full/requirements_cuda128.txt        | 2 +-
 requirements/full/requirements_cuda128_noavx2.txt | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements/full/requirements_cuda128.txt b/requirements/full/requirements_cuda128.txt
index bf9ee6cb..f8dca352 100644
--- a/requirements/full/requirements_cuda128.txt
+++ b/requirements/full/requirements_cuda128.txt
@@ -24,7 +24,7 @@ scipy
 sentencepiece
 tensorboard
 transformers==4.53.*
-triton-windows==3.2.0.post19; platform_system == "Windows"
+triton-windows==3.3.1.post19; platform_system == "Windows"
 tqdm
 wandb
 
diff --git a/requirements/full/requirements_cuda128_noavx2.txt b/requirements/full/requirements_cuda128_noavx2.txt
index 691f2ca4..a3860b6a 100644
--- a/requirements/full/requirements_cuda128_noavx2.txt
+++ b/requirements/full/requirements_cuda128_noavx2.txt
@@ -24,7 +24,7 @@ scipy
 sentencepiece
 tensorboard
 transformers==4.53.*
-triton-windows==3.2.0.post19; platform_system == "Windows"
+triton-windows==3.3.1.post19; platform_system == "Windows"
 tqdm
 wandb