From 21e0e9f32bf7ed7989538c8b8f0755ae3f4d57c9 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Wed, 9 Jul 2025 21:05:17 -0700 Subject: [PATCH] Add the triton-windows requirement on Windows to make transformers functional --- requirements/full/requirements.txt | 1 + requirements/full/requirements_amd.txt | 1 + requirements/full/requirements_amd_noavx2.txt | 1 + requirements/full/requirements_apple_intel.txt | 1 + requirements/full/requirements_apple_silicon.txt | 1 + requirements/full/requirements_cpu_only.txt | 1 + requirements/full/requirements_cpu_only_noavx2.txt | 1 + requirements/full/requirements_cuda128.txt | 1 + requirements/full/requirements_cuda128_noavx2.txt | 1 + requirements/full/requirements_noavx2.txt | 1 + requirements/full/requirements_nowheels.txt | 1 + 11 files changed, 11 insertions(+) diff --git a/requirements/full/requirements.txt b/requirements/full/requirements.txt index 791a45eb..49000383 100644 --- a/requirements/full/requirements.txt +++ b/requirements/full/requirements.txt @@ -24,6 +24,7 @@ scipy sentencepiece tensorboard transformers==4.53.* +triton-windows==3.3.1.post19; platform_system == "Windows" tqdm wandb diff --git a/requirements/full/requirements_amd.txt b/requirements/full/requirements_amd.txt index af117c5c..b72b6806 100644 --- a/requirements/full/requirements_amd.txt +++ b/requirements/full/requirements_amd.txt @@ -23,6 +23,7 @@ scipy sentencepiece tensorboard transformers==4.53.* +triton-windows==3.3.1.post19; platform_system == "Windows" tqdm wandb diff --git a/requirements/full/requirements_amd_noavx2.txt b/requirements/full/requirements_amd_noavx2.txt index 8095c38c..898f3451 100644 --- a/requirements/full/requirements_amd_noavx2.txt +++ b/requirements/full/requirements_amd_noavx2.txt @@ -23,6 +23,7 @@ scipy sentencepiece tensorboard transformers==4.53.* +triton-windows==3.3.1.post19; platform_system == "Windows" tqdm wandb diff --git a/requirements/full/requirements_apple_intel.txt b/requirements/full/requirements_apple_intel.txt index d63207b6..926baaf0 100644 --- a/requirements/full/requirements_apple_intel.txt +++ b/requirements/full/requirements_apple_intel.txt @@ -23,6 +23,7 @@ scipy sentencepiece tensorboard transformers==4.53.* +triton-windows==3.3.1.post19; platform_system == "Windows" tqdm wandb diff --git a/requirements/full/requirements_apple_silicon.txt b/requirements/full/requirements_apple_silicon.txt index a2add551..b1fbebd6 100644 --- a/requirements/full/requirements_apple_silicon.txt +++ b/requirements/full/requirements_apple_silicon.txt @@ -23,6 +23,7 @@ scipy sentencepiece tensorboard transformers==4.53.* +triton-windows==3.3.1.post19; platform_system == "Windows" tqdm wandb diff --git a/requirements/full/requirements_cpu_only.txt b/requirements/full/requirements_cpu_only.txt index ec30e05d..7cb2b2da 100644 --- a/requirements/full/requirements_cpu_only.txt +++ b/requirements/full/requirements_cpu_only.txt @@ -23,6 +23,7 @@ scipy sentencepiece tensorboard transformers==4.53.* +triton-windows==3.3.1.post19; platform_system == "Windows" tqdm wandb diff --git a/requirements/full/requirements_cpu_only_noavx2.txt b/requirements/full/requirements_cpu_only_noavx2.txt index d7cf75db..1aa90fbd 100644 --- a/requirements/full/requirements_cpu_only_noavx2.txt +++ b/requirements/full/requirements_cpu_only_noavx2.txt @@ -23,6 +23,7 @@ scipy sentencepiece tensorboard transformers==4.53.* +triton-windows==3.3.1.post19; platform_system == "Windows" tqdm wandb diff --git a/requirements/full/requirements_cuda128.txt b/requirements/full/requirements_cuda128.txt index dd5c08ac..2f52baf9 100644 --- a/requirements/full/requirements_cuda128.txt +++ b/requirements/full/requirements_cuda128.txt @@ -24,6 +24,7 @@ scipy sentencepiece tensorboard transformers==4.53.* +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 17dd31fd..d06a235e 100644 --- a/requirements/full/requirements_cuda128_noavx2.txt +++ b/requirements/full/requirements_cuda128_noavx2.txt @@ -24,6 +24,7 @@ scipy sentencepiece tensorboard transformers==4.53.* +triton-windows==3.3.1.post19; platform_system == "Windows" tqdm wandb diff --git a/requirements/full/requirements_noavx2.txt b/requirements/full/requirements_noavx2.txt index be914200..a5231cde 100644 --- a/requirements/full/requirements_noavx2.txt +++ b/requirements/full/requirements_noavx2.txt @@ -24,6 +24,7 @@ scipy sentencepiece tensorboard transformers==4.53.* +triton-windows==3.3.1.post19; platform_system == "Windows" tqdm wandb diff --git a/requirements/full/requirements_nowheels.txt b/requirements/full/requirements_nowheels.txt index 4f35ffcc..e4458ff6 100644 --- a/requirements/full/requirements_nowheels.txt +++ b/requirements/full/requirements_nowheels.txt @@ -23,6 +23,7 @@ scipy sentencepiece tensorboard transformers==4.53.* +triton-windows==3.3.1.post19; platform_system == "Windows" tqdm wandb