Update Gemma 2 Notebook (#32200)

apache · Aug 16, 2024 · ff93c48 · ff93c48
1 parent 65550a7
commit ff93c48
Showing 1 changed file with 4 additions and 11 deletions.
diff --git a/examples/notebooks/beam-ml/gemma_2_sentiment_and_summarization.ipynb b/examples/notebooks/beam-ml/gemma_2_sentiment_and_summarization.ipynb
@@ -293,7 +293,7 @@
       },
       "outputs": [],
       "source": [
-        "pip install apache_beam[gcp]==\"2.54.0\" keras_nlp==\"0.14.3\" keras>=\"3\" jax[cuda12]"
+        "%pip install apache_beam[gcp]==\"2.54.0\" keras_nlp==\"0.14.3\" keras==\"3.5.0\" jax[cuda12]"
       ]
     },
     {
@@ -596,16 +596,9 @@
     "colab": {
       "provenance": []
     },
-    "environment": {
-      "kernel": "apache-beam-2.57.0",
-      "name": ".m121",
-      "type": "gcloud",
-      "uri": "us-docker.pkg.dev/deeplearning-platform-release/gcr.io/:m121"
-    },
     "kernelspec": {
-      "display_name": "Apache Beam 2.57.0 (Local)",
-      "language": "python",
-      "name": "apache-beam-2.57.0"
+      "display_name": "Python 3",
+      "name": "python3"
     },
     "language_info": {
       "codemirror_mode": {
@@ -617,7 +610,7 @@
       "name": "python",
       "nbconvert_exporter": "python",
       "pygments_lexer": "ipython3",
-      "version": "3.10.14"
+      "version": "3.10.1"
     }
   },
   "nbformat": 4,