diff --git a/docs/tutorials/transformer.ipynb b/docs/tutorials/transformer.ipynb
index 1a7c5ef6e..7a0d39708 100644
--- a/docs/tutorials/transformer.ipynb
+++ b/docs/tutorials/transformer.ipynb
@@ -225,6 +225,7 @@
         "# masking support for `tf.keras.layers.MultiHeadAttention`.\n",
         "!apt install --allow-change-held-packages libcudnn8=8.1.0.77-1+cuda11.2\n",
         "!pip uninstall -y -q tensorflow keras tensorflow-estimator tensorflow-text\n",
+        "!pip install protobuf~=3.20.3\n",
         "!pip install -q tensorflow_datasets\n",
         "!pip install -q -U tensorflow-text tensorflow"
       ]