diff --git a/requirements/full/requirements_cuda128.txt b/requirements/full/requirements_cuda128.txt index 801009fb..7851041f 100644 --- a/requirements/full/requirements_cuda128.txt +++ b/requirements/full/requirements_cuda128.txt @@ -24,8 +24,7 @@ safetensors==0.5.* scipy sentencepiece tensorboard -transformers==4.52.* -triton-windows; platform_system == "Windows" +transformers==4.50.* tqdm wandb diff --git a/requirements/full/requirements_cuda128_noavx2.txt b/requirements/full/requirements_cuda128_noavx2.txt index 8b77e144..c8015166 100644 --- a/requirements/full/requirements_cuda128_noavx2.txt +++ b/requirements/full/requirements_cuda128_noavx2.txt @@ -24,8 +24,7 @@ safetensors==0.5.* scipy sentencepiece tensorboard -transformers==4.52.* -triton-windows; platform_system == "Windows" +transformers==4.50.* tqdm wandb