diff --git a/sdks/python/apache_beam/ml/inference/huggingface_tests_requirements.txt b/sdks/python/apache_beam/ml/inference/huggingface_tests_requirements.txt index 9b9e9bdd55f1..704b29a76ea5 100644 --- a/sdks/python/apache_beam/ml/inference/huggingface_tests_requirements.txt +++ b/sdks/python/apache_beam/ml/inference/huggingface_tests_requirements.txt @@ -16,5 +16,5 @@ # torch>=1.7.1 -transformers==4.53.0 +transformers==5.5.0 tensorflow>=2.12.0 \ No newline at end of file diff --git a/sdks/python/apache_beam/ml/inference/torch_tests_requirements.txt b/sdks/python/apache_beam/ml/inference/torch_tests_requirements.txt index df6273038f1a..ef816ae159ac 100644 --- a/sdks/python/apache_beam/ml/inference/torch_tests_requirements.txt +++ b/sdks/python/apache_beam/ml/inference/torch_tests_requirements.txt @@ -18,5 +18,5 @@ torch>=1.7.1 torchvision>=0.8.2 pillow>=8.0.0 -transformers>=4.18.0 +transformers>=5.5.0 google-cloud-monitoring>=2.27.0 \ No newline at end of file diff --git a/sdks/python/apache_beam/ml/inference/vllm_tests_requirements.txt b/sdks/python/apache_beam/ml/inference/vllm_tests_requirements.txt index ad5877edbec5..ea1f2550c18a 100644 --- a/sdks/python/apache_beam/ml/inference/vllm_tests_requirements.txt +++ b/sdks/python/apache_beam/ml/inference/vllm_tests_requirements.txt @@ -17,7 +17,7 @@ torch>=1.7.1 torchvision>=0.8.2 pillow>=8.0.0 -transformers==4.57.1 +transformers==5.5.0 sentencepiece==0.2.1 google-cloud-monitoring>=2.27.0 openai>=1.52.2