From 702333977f7add739023da9ac946c412c27b31fa Mon Sep 17 00:00:00 2001 From: Wei Wei <5577741+windmaple@users.noreply.github.com> Date: Tue, 11 Jun 2024 23:49:19 +0800 Subject: [PATCH] Update distributed_tuning.ipynb (#355) 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 36b961c72..ea5a355e1 100644 --- a/site/en/gemma/docs/distributed_tuning.ipynb +++ b/site/en/gemma/docs/distributed_tuning.ipynb @@ -204,7 +204,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" ]