diff --git a/requirements/requirements_lightning.txt b/requirements/requirements_lightning.txt index c6bd2950b8f8..1c8416da287b 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~=5.3.0 +transformers~=5.5.0 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 ddfa4deaf19d..8f967ef90dbb 100644 --- a/requirements/requirements_vllm.txt +++ b/requirements/requirements_vllm.txt @@ -20,7 +20,7 @@ pangu rouge_score sacrebleu scikit-learn -transformers==5.3.0 +transformers==5.5.0 vllm==0.22.0 webdataset>=0.2.86 wget