From 24ea1e3faa27b2ac179be44e6cf2cd04990104dd Mon Sep 17 00:00:00 2001 From: Robert Neale Date: Tue, 28 Mar 2023 14:08:43 -0700 Subject: [PATCH] Pin protobuf version to prevent failure. See https://github.com/tensorflow/text/issues/1115 for more info. PiperOrigin-RevId: 520129712 --- docs/tutorials/transformer.ipynb | 1 + 1 file changed, 1 insertion(+) 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" ]