From 6d33b8c5db0eb0f2ab162bd5a2f51a5fde6f6971 Mon Sep 17 00:00:00 2001
From: Rick van Hattem <wolph@wol.ph>
Date: Tue, 20 Feb 2024 03:19:01 +0100
Subject: [PATCH] Removed install triton when running on WIndows

---
 requirements/pt2.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements/pt2.txt b/requirements/pt2.txt
index 43fd62652..2d1696102 100644
--- a/requirements/pt2.txt
+++ b/requirements/pt2.txt
@@ -31,7 +31,7 @@ torchmetrics>=1.0.1
 torchvision>=0.15.2
 tqdm>=4.65.0
 transformers==4.19.1
-triton==2.0.0
+triton==2.0.0; platform_system != 'Windows'
 urllib3<1.27,>=1.25.4
 wandb>=0.15.6
 webdataset>=0.2.33