From ff93c48a4c571607811c1028b4d05a44da1859ee Mon Sep 17 00:00:00 2001
From: Jack McCluskey <34928439+jrmccluskey@users.noreply.github.com>
Date: Fri, 16 Aug 2024 05:12:57 -0400
Subject: [PATCH] Update Gemma 2 Notebook (#32200)

---
 .../gemma_2_sentiment_and_summarization.ipynb     | 15 ++++-----------
 1 file changed, 4 insertions(+), 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
index 61bba5f10cc0..64151d29992d 100644
--- 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,