Code update

PiperOrigin-RevId: 811527180

Author: sizhit2

examples/qlora_demo.ipynb

@@ -764,15 +764,15 @@
       ]
     },
     {
-      "cell_type": "markdown",
       "metadata": {
-        "id": "eIw2rIKmT3Uv"
+        "id": "CrZnJyQP87eM"
       },
+      "cell_type": "markdown",
       "source": [
-        "\u003cfont size=3\u003eSetup\u003cfont\u003e           | \u003cfont size=3\u003eTrain Step Time\u003cfont\u003e | \u003cfont size=3\u003ePeak Memory Usage\u003cfont\u003e\n",
-        "---------------------------------- | ---------------------------------- | ------------------------------\n",
-        "\u003cfont size=3\u003eFull weights\u003cfont\u003e        |   \u003cfont size=3\u003e~1.22 s\u003cfont\u003e     |   \u003cfont size=3\u003e43.26 GiB\u003cfont\u003e\n",
-        "\u003cfont size=3\u003eQLoRA\u003cfont\u003e        |   \u003cfont size=3\u003e~1.19 s\u003cfont\u003e     |   \u003cfont size=3\u003e28.14 GiB\u003cfont\u003e"
+        "| Setup        | Train Step Time | Peak Memory Usage |\n",
+        "|--------------|----------------:|------------------:|\n",
+        "| Full weights |        ~1.22 s  |         43.26 GiB |\n",
+        "| QLoRA        |        ~1.19 s  |         28.14 GiB |"
       ]
     },
     {