Fix bug in smoking cessation example

PiperOrigin-RevId: 636854181 Change-Id: I5e03eaa494f135e4ecf9fc284f77a301c85a4091
google-deepmind · May 24, 2024 · 13cb096 · 13cb096
1 parent ead2d77
commit 13cb096
Showing 1 changed file with 3 additions and 23 deletions.
diff --git a/examples/smoking_cessation_in_an_alchemical_world.ipynb b/examples/smoking_cessation_in_an_alchemical_world.ipynb
@@ -24,7 +24,7 @@
         "id": "J2TwJrZ08wXz"
       },
       "source": [
-        "## Setup and imports"
+        "## Install packages and imports"
       ]
     },
     {
@@ -153,33 +153,13 @@
         "embedder = lambda x: st_model.encode(x, show_progress_bar=False)"
       ]
     },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "Yc1QsPHcwAaI"
-      },
-      "outputs": [],
-      "source": [
-        "# @title SAX Language Model\n",
-        "\n",
-        "SAX_PATH = '/sax/simulacra/llmit-340b-v10'  # @param [\"/sax/simulacra/llmit-340b-v10-8k\", \"/sax/simulacra/llmit-340b-v10\"]\n",
-        "DEFAULT_MAX_TOKENS = 300  # @param {type: 'integer'}\n",
-        "DEFAULT_TIMEOUT_SECONDS = 60  # @param {type: 'number'}\n",
-        "\n",
-        "model = sax_model.SAXLanguageModel(\n",
-        "    SAX_PATH,\n",
-        "    deterministic_multiple_choice=True,\n",
-        ")"
-      ]
-    },
     {
       "cell_type": "markdown",
       "metadata": {
-        "id": "z9HYjZgyakc_"
+        "id": "ZP49rm1dC92V"
       },
       "source": [
-        "## Configuring the generic knowledge of players and GM."
+        "# Setup"
       ]
     },
     {