diff --git a/dataflow/gemma-flex-template/requirements-test.txt b/dataflow/gemma-flex-template/requirements-test.txt
index 5e6dcfc99aa..f6bb1917e2f 100644
--- a/dataflow/gemma-flex-template/requirements-test.txt
+++ b/dataflow/gemma-flex-template/requirements-test.txt
@@ -1,4 +1,4 @@
-google-cloud-aiplatform==1.62.0
+google-cloud-aiplatform==1.133.0
 google-cloud-dataflow-client==0.8.14
 google-cloud-pubsub==2.28.0
 google-cloud-storage==2.18.2
diff --git a/dataflow/run-inference/requirements-test.txt b/dataflow/run-inference/requirements-test.txt
index c9095c832fd..3f2c86fa103 100644
--- a/dataflow/run-inference/requirements-test.txt
+++ b/dataflow/run-inference/requirements-test.txt
@@ -1,4 +1,4 @@
-google-cloud-aiplatform==1.57.0
+google-cloud-aiplatform==1.133.0
 google-cloud-dataflow-client==0.8.14
 google-cloud-storage==2.10.0
 pytest==8.2.0
diff --git a/gemma2/requirements.txt b/gemma2/requirements.txt
index f8990233d3f..746caf678ab 100644
--- a/gemma2/requirements.txt
+++ b/gemma2/requirements.txt
@@ -1,2 +1,2 @@
-google-cloud-aiplatform[all]==1.64.0
+google-cloud-aiplatform[all]==1.133.0
 protobuf==5.29.5