Update example.ipynb

Author: shauli-ravfogel

example.ipynb

@@ -26,14 +26,6 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "/cluster/home/sravfogel/venv/lib/python3.12/site-packages/huggingface_hub/file_download.py:1150: FutureWarning: `resume_download` is deprecated and will be removed in version 1.0.0. Downloads always resume when possible. If you want to force a new download, use `force_download=True`.\n",
-      "  warnings.warn(\n",
-      "/cluster/home/sravfogel/venv/lib/python3.12/site-packages/huggingface_hub/file_download.py:1150: FutureWarning: `resume_download` is deprecated and will be removed in version 1.0.0. Downloads always resume when possible. If you want to force a new download, use `force_download=True`.\n",
-      "  warnings.warn(\n",
-      "/cluster/home/sravfogel/venv/lib/python3.12/site-packages/huggingface_hub/file_download.py:1150: FutureWarning: `resume_download` is deprecated and will be removed in version 1.0.0. Downloads always resume when possible. If you want to force a new download, use `force_download=True`.\n",
-      "  warnings.warn(\n",
-      "/cluster/home/sravfogel/venv/lib/python3.12/site-packages/huggingface_hub/file_download.py:1150: FutureWarning: `resume_download` is deprecated and will be removed in version 1.0.0. Downloads always resume when possible. If you want to force a new download, use `force_download=True`.\n",
-      "  warnings.warn(\n"
      ]
     }
    ],
@@ -53,20 +45,6 @@
     "                                                       use_fast=False,trust_remote_code=True)"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 104,
-   "id": "eb7b2def-4d08-44fa-8bbf-75fdee14ce5c",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "prompt = \"the louvre is one of the most known museums in the city of\"\n",
-    "# Sample a continuation from the original model\n",
-    "num_new_tokens = 40\n",
-    "utils.set_seed(0)\n",
-    "original_continuation_tokens, original_continuation = utils.get_continuation(original_model, tokenizer, prompt, max_new_tokens=num_new_tokens, return_only_continuation=True,num_beams=1, do_sample=True, token_healing=True)"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 112,