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