reverse order of import and backend steps

google · Mar 17, 2024 · b1cfa24 · b1cfa24
1 parent 8347e6f
commit b1cfa24
Showing 1 changed file with 13 additions and 17 deletions.
diff --git a/site/en/gemma/docs/get_started.ipynb b/site/en/gemma/docs/get_started.ipynb
@@ -148,48 +148,44 @@
     {
       "cell_type": "markdown",
       "metadata": {
-        "id": "FX47AUYrXwLK"
+        "id": "Pm5cVOFt5YvZ"
       },
       "source": [
-        "### Import packages\n",
+        "### Select a backend\n",
         "\n",
-        "Import Keras and KerasNLP."
+        "Keras is a high-level, multi-framework deep learning API designed for simplicity and ease of use. [Keras 3](https://keras.io/keras_3) lets you choose the backend: TensorFlow, JAX, or PyTorch. All three will work for this tutorial."
       ]
     },
     {
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "ww83zI9ToPso"
+        "id": "7rS7ryTs5wjf"
       },
       "outputs": [],
       "source": [
-        "import keras\n",
-        "import keras_nlp"
+        "import os\n",
+        "\n",
+        "os.environ[\"KERAS_BACKEND\"] = \"jax\"  # Or \"tensorflow\" or \"torch\"."
       ]
     },
     {
       "cell_type": "markdown",
-      "metadata": {
-        "id": "Pm5cVOFt5YvZ"
-      },
+      "metadata": {},
       "source": [
-        "### Select a backend\n",
+        "### Import packages\n",
         "\n",
-        "Keras is a high-level, multi-framework deep learning API designed for simplicity and ease of use. [Keras 3](https://keras.io/keras_3) lets you choose the backend: TensorFlow, JAX, or PyTorch. All three will work for this tutorial."
+        "Import Keras and KerasNLP."
       ]
     },
     {
       "cell_type": "code",
       "execution_count": null,
-      "metadata": {
-        "id": "7rS7ryTs5wjf"
-      },
+      "metadata": {},
       "outputs": [],
       "source": [
-        "import os\n",
-        "\n",
-        "os.environ[\"KERAS_BACKEND\"] = \"jax\"  # Or \"tensorflow\" or \"torch\"."
+        "import keras\n",
+        "import keras_nlp"
       ]
     },
     {