Updated walkthrough notebook

Signed-off-by: Grace Sng <[email protected]>

Author: grace-sng7

notebooks/walkthrough.ipynb

@@ -238,6 +238,52 @@
         }
       ]
     },
+    {
+      "cell_type": "code",
+      "source": [
+        "pip install gpustat"
+      ],
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "pn7WLx36DpTg",
+        "outputId": "8413ae67-ab22-4954-d671-4ddd4cc22381"
+      },
+      "execution_count": 23,
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "Collecting gpustat\n",
+            "  Downloading gpustat-1.1.1.tar.gz (98 kB)\n",
+            "\u001b[?25l     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m0.0/98.1 kB\u001b[0m \u001b[31m?\u001b[0m eta \u001b[36m-:--:--\u001b[0m\r\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m98.1/98.1 kB\u001b[0m \u001b[31m9.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+            "\u001b[?25h  Installing build dependencies ... \u001b[?25l\u001b[?25hdone\n",
+            "  Getting requirements to build wheel ... \u001b[?25l\u001b[?25hdone\n",
+            "  Preparing metadata (pyproject.toml) ... \u001b[?25l\u001b[?25hdone\n",
+            "Collecting nvidia-ml-py>=11.450.129 (from gpustat)\n",
+            "  Downloading nvidia_ml_py-12.570.86-py3-none-any.whl.metadata (8.7 kB)\n",
+            "Requirement already satisfied: psutil>=5.6.0 in /usr/local/lib/python3.11/dist-packages (from gpustat) (5.9.5)\n",
+            "Collecting blessed>=1.17.1 (from gpustat)\n",
+            "  Downloading blessed-1.20.0-py2.py3-none-any.whl.metadata (13 kB)\n",
+            "Requirement already satisfied: wcwidth>=0.1.4 in /usr/local/lib/python3.11/dist-packages (from blessed>=1.17.1->gpustat) (0.2.13)\n",
+            "Requirement already satisfied: six>=1.9.0 in /usr/local/lib/python3.11/dist-packages (from blessed>=1.17.1->gpustat) (1.17.0)\n",
+            "Downloading blessed-1.20.0-py2.py3-none-any.whl (58 kB)\n",
+            "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m58.4/58.4 kB\u001b[0m \u001b[31m5.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+            "\u001b[?25hDownloading nvidia_ml_py-12.570.86-py3-none-any.whl (44 kB)\n",
+            "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m44.4/44.4 kB\u001b[0m \u001b[31m4.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+            "\u001b[?25hBuilding wheels for collected packages: gpustat\n",
+            "  Building wheel for gpustat (pyproject.toml) ... \u001b[?25l\u001b[?25hdone\n",
+            "  Created wheel for gpustat: filename=gpustat-1.1.1-py3-none-any.whl size=26608 sha256=56886eeb00c620277ad48366e6f60b747bd7cd9ebb3da078c06e8b1eb030a6a6\n",
+            "  Stored in directory: /root/.cache/pip/wheels/c9/2b/d9/a0b77d6e8623ce6b5c73813af455a3ace394abfc2e8aef7ed6\n",
+            "Successfully built gpustat\n",
+            "Installing collected packages: nvidia-ml-py, blessed, gpustat\n",
+            "Successfully installed blessed-1.20.0 gpustat-1.1.1 nvidia-ml-py-12.570.86\n"
+          ]
+        }
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": null,
@@ -262,23 +308,11 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 8,
+      "execution_count": 24,
       "metadata": {
-        "id": "GtLY4ihkJ5YR",
-        "colab": {
-          "base_uri": "https://localhost:8080/"
-        },
-        "outputId": "1e1e579c-c9fc-4fcc-fc97-f26821844b62"
+        "id": "GtLY4ihkJ5YR"
       },
-      "outputs": [
-        {
-          "output_type": "stream",
-          "name": "stderr",
-          "text": [
-            "WARNING:guidance.models._model:gpustat is not installed, run `pip install gpustat` to collect GPU stats.\n"
-          ]
-        }
-      ],
+      "outputs": [],
       "source": [
         "from pywhyllm.suggesters.simple_model_suggester import SimpleModelSuggester\n",
         "modeler = SimpleModelSuggester('gpt-4')"
@@ -538,23 +572,11 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 18,
+      "execution_count": 25,
       "metadata": {
-        "id": "MNL9qLCNJ5YT",
-        "colab": {
-          "base_uri": "https://localhost:8080/"
-        },
-        "outputId": "bcb1f6e0-2937-439b-a3f0-0f4ec91567e6"
+        "id": "MNL9qLCNJ5YT"
       },
-      "outputs": [
-        {
-          "output_type": "stream",
-          "name": "stderr",
-          "text": [
-            "WARNING:guidance.models._model:gpustat is not installed, run `pip install gpustat` to collect GPU stats.\n"
-          ]
-        }
-      ],
+      "outputs": [],
       "source": [
         "from pywhyllm.suggesters.simple_identification_suggester import SimpleIdentificationSuggester\n",
         "identifier = SimpleIdentificationSuggester('gpt-4')"