diff --git a/requirements/requirements_lightning.txt b/requirements/requirements_lightning.txt index b2faa2680cf8..c6bd2950b8f8 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.0.0rc3 +transformers~=5.3.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 7bc66df7dbe8..ddfa4deaf19d 100644 --- a/requirements/requirements_vllm.txt +++ b/requirements/requirements_vllm.txt @@ -20,7 +20,7 @@ pangu rouge_score sacrebleu scikit-learn -transformers==5.0.0rc3 +transformers==5.3.0 vllm==0.22.0 webdataset>=0.2.86 wget