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