diff --git a/requirements/requirements_lightning.txt b/requirements/requirements_lightning.txt index 10be5ca2954c..b2faa2680cf8 100644 --- a/requirements/requirements_lightning.txt +++ b/requirements/requirements_lightning.txt @@ -5,7 +5,7 @@ lightning>2.2.1,<=2.4.0 omegaconf<=2.3 peft torchmetrics>=0.11.0 -transformers~=4.53.0 +transformers~=5.0.0rc3 wandb webdataset>=0.2.86 nv_one_logger_core>=2.1.0 diff --git a/requirements/requirements_vllm.txt b/requirements/requirements_vllm.txt index 01ab8adcc227..376ccc143b6e 100644 --- a/requirements/requirements_vllm.txt +++ b/requirements/requirements_vllm.txt @@ -20,7 +20,7 @@ pangu rouge_score sacrebleu scikit-learn -transformers==4.53.0 +transformers==5.0.0rc3 vllm==0.11.0 webdataset>=0.2.86 wget