Update distributed_tuning.ipynb (google#355)

windmaple · web-flow · commit 702333977f7a · 2024-06-11T08:49:19.000-07:00
Add pip install tf-text; otherwise it errors out when loading Gemma
diff --git 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"
       ]

Original file line number	Diff line number	Diff line change
`@@ -204,7 +204,7 @@`
`204`	`204`	`"# Work around an import error with tensorflow-hub. The library is not used.\n",`
`205`	`205`	`"!pip install -q -U tensorflow-hub\n",`
`206`	`206`	`"# Install tensorflow-cpu so tensorflow does not attempt to access the TPU.\n",`
`207`		`- "!pip install -q -U tensorflow-cpu\n",`
	`207`	`+ "!pip install -q -U tensorflow-cpu tensorflow-text\n",`
`208`	`208`	`"# Install keras 3 last. See https://keras.io/getting_started for details.\n",`
`209`	`209`	`"!pip install -q -U keras"`
`210`	`210`	`]`