diff --git a/llm_on_ray/pretrain/requirements.optimum-habana.txt b/llm_on_ray/pretrain/requirements.optimum-habana.txt index a7504978..88aa071e 100644 --- a/llm_on_ray/pretrain/requirements.optimum-habana.txt +++ b/llm_on_ray/pretrain/requirements.optimum-habana.txt @@ -2,7 +2,7 @@ accelerate==0.21.0 datasets==2.12.0 numpy==1.24.4 https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-3.0.0.dev0-cp38-cp38-manylinux2014_x86_64.whl -transformers==4.50.0 +transformers==4.52.1 typing==3.7.4.3 tabulate ray[tune] diff --git a/llm_on_ray/pretrain/requirements.txt b/llm_on_ray/pretrain/requirements.txt index 8cddb582..d5c5482e 100644 --- a/llm_on_ray/pretrain/requirements.txt +++ b/llm_on_ray/pretrain/requirements.txt @@ -4,7 +4,7 @@ numpy==1.24.4 https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-3.0.0.dev0-cp38-cp38-manylinux2014_x86_64.whl torchvision==0.14.1 torch -transformers==4.50.0 +transformers==4.52.1 typing==3.7.4.3 tabulate ray[tune]