Don't downgrade triton-windows on CUDA 12.8

This commit is contained in:
oobabooga 2025-07-10 05:39:04 -07:00
parent e523f25b9f
commit 0f3a88057c
2 changed files with 2 additions and 2 deletions

View file

@ -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

View file

@ -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