Do not bump Transformers to 4.52 on CUDA 12.8

Performance is slow, and the older version works fine with torch 2.7.
This commit is contained in:
oobabooga 2025-06-10 18:45:42 -07:00
parent bc921c66e5
commit 552cb09f09
2 changed files with 2 additions and 4 deletions

View file

@ -24,8 +24,7 @@ safetensors==0.5.*
scipy
sentencepiece
tensorboard
transformers==4.52.*
triton-windows; platform_system == "Windows"
transformers==4.50.*
tqdm
wandb

View file

@ -24,8 +24,7 @@ safetensors==0.5.*
scipy
sentencepiece
tensorboard
transformers==4.52.*
triton-windows; platform_system == "Windows"
transformers==4.50.*
tqdm
wandb