From 38d714b8fe7f3a57d0a807cfe968ed998ccdb430 Mon Sep 17 00:00:00 2001 From: Wayne Wei Date: Sun, 7 Apr 2024 14:55:38 +0800 Subject: [PATCH] Update distributed_tuning.ipynb Add pip install tf-text; otherwise it errors out when loading Gemma --- site/en/gemma/docs/distributed_tuning.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/site/en/gemma/docs/distributed_tuning.ipynb b/site/en/gemma/docs/distributed_tuning.ipynb index 30b8d99fc..77cb61d61 100644 --- a/site/en/gemma/docs/distributed_tuning.ipynb +++ b/site/en/gemma/docs/distributed_tuning.ipynb @@ -200,7 +200,7 @@ "# Work around an import error with tensorflow-hub. The library is not used.\n", "!pip install -q -U tensorflow-hub\n", "# Install tensorflow-cpu so tensorflow does not attempt to access the TPU.\n", - "!pip install -q -U tensorflow-cpu\n", + "!pip install -q -U tensorflow-cpu tensorflow-text\n", "# Install keras 3 last. See https://keras.io/getting_started for details.\n", "!pip install -q -U keras" ]