Sync docs from Google3 to GitHub

google · Feb 28, 2024 · 4a775b3 · 4a775b3
1 parent f28d6d3
commit 4a775b3
Show file tree

Hide file tree

Showing 3 changed files with 134 additions and 61 deletions.
diff --git a/site/en/gemma/docs/agile_classifiers.ipynb b/site/en/gemma/docs/agile_classifiers.ipynb
@@ -131,7 +131,8 @@
       },
       "outputs": [],
       "source": [
-        "!pip install keras keras-nlp"
+        "!pip install -q -U keras-nlp\n",
+        "!pip install -q -U keras"
       ]
     },
     {
@@ -1089,7 +1090,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 1,
+      "execution_count": null,
       "metadata": {
         "id": "fMSDQtyTgeP0"
       },

diff --git a/site/en/gemma/docs/distributed_tuning.ipynb b/site/en/gemma/docs/distributed_tuning.ipynb
@@ -252,10 +252,13 @@
         }
       ],
       "source": [
-        "# Install Keras 3 last. See https://keras.io/getting_started/ for more details.\n",
-        "!pip install -q tensorflow-cpu\n",
-        "!pip install -q -U keras-nlp tensorflow-hub\n",
-        "!pip install -q -U keras>=3"
+        "!pip install -q -U keras-nlp\n",
+        "# 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",
+        "# Install keras 3 last. See https://keras.io/getting_started for details.\n",
+        "!pip install -q -U keras"
       ]
     },
     {