diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini
index 8cc125d946a8..8cdc4a98bbfe 100644
--- a/sdks/python/tox.ini
+++ b/sdks/python/tox.ini
@@ -394,10 +394,12 @@ commands =
 
 [testenv:py39-tensorflow-212]
 deps =
-  212: tensorflow>=2.12rc1,<2.13
-  # Help pip resolve conflict with typing-extensions for old version of TF https://github.com/apache/beam/issues/30852
-  212: pydantic<2.7
-extras = test,gcp
+  212: 
+    tensorflow>=2.12rc1,<2.13
+    # Help pip resolve conflict with typing-extensions for old version of TF https://github.com/apache/beam/issues/30852
+    pydantic<2.7
+    protobuf==4.25.5
+extras = test,gcp,ml_test
 commands =
   # Log tensorflow version for debugging
   /bin/sh -c "pip freeze | grep -E tensorflow"
@@ -428,7 +430,8 @@ deps =
   430: transformers>=4.30.0,<4.31.0
   torch>=1.9.0,<1.14.0
   tensorflow==2.12.0
-extras = test,gcp
+  protobuf==4.25.5
+extras = test,gcp,ml_test
 commands =
   # Log transformers and its dependencies version for debugging
   /bin/sh -c "pip freeze | grep -E transformers"