feat: Update training notebook

decodingml · Jan 29, 2025 · 6cfa12d · 6cfa12d
1 parent 5c3bc8e
commit 6cfa12d
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/apps/second-brain-offline/src/second_brain_offline/application/models/finetuning.ipynb b/apps/second-brain-offline/src/second_brain_offline/application/models/finetuning.ipynb
@@ -65,6 +65,33 @@
         "# !pip install --force-reinstall --no-cache-dir --no-deps git+https://github.com/unslothai/unsloth.git"
       ]
     },
+    {
+      "cell_type": "markdown",
+      "metadata": {},
+      "source": [
+        "### Prepare environment"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "import os\n",
+        "\n",
+        "from getpass import getpass\n",
+        "\n",
+        "hf_token = getpass(\"Enter your Hugging Face token: \")\n",
+        "\n",
+        "comet_api_key = getpass(\"Enter your Comet API key: \")\n",
+        "comet_project_name = \"second-brain-course\"\n",
+        "\n",
+        "os.environ[\"HF_TOKEN\"] = hf_token\n",
+        "os.environ[\"COMET_API_KEY\"] = comet_api_key\n",
+        "os.environ[\"COMET_PROJECT_NAME\"] = comet_project_name"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {