Fix tutorial error

Author: taufeeque9

tutorials/code_intervention.ipynb

@@ -15,12 +15,15 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "! pip install codebook-features\n",
-    "# restart runtime after installing if running on Colab\n",
-    "from IPython import get_ipython\n",
+    "try:\n",
+    "    import codebook_features\n",
+    "except:\n",
+    "    ! pip install codebook-features\n",
+    "    # restart runtime after installing if running on Colab\n",
+    "    from IPython import get_ipython\n",
     "\n",
-    "if \"google.colab\" in str(get_ipython()):\n",
-    "    exit(0)"
+    "    if \"google.colab\" in str(get_ipython()):\n",
+    "        exit(0)"
    ]
   },
   {
@@ -106,6 +109,7 @@
    "source": [
     "sentence = \"this is a random sentence to test.\"\n",
     "input_tensor = tokenizer(sentence, return_tensors=\"pt\")[\"input_ids\"]\n",
+    "input_tensor = input_tensor.to(device)\n",
     "output = orig_cb_model(input_tensor)[\"logits\"]\n",
     "hooked_output = cb_model(input_tensor)\n",
     "assert torch.allclose(output, hooked_output, atol=1e-4)"