diff --git a/demo.ipynb b/demo.ipynb
index adb0d08..d0a1b10 100644
--- a/demo.ipynb
+++ b/demo.ipynb
@@ -500,6 +500,9 @@
     "    \"\"\"\n",
     "    Extracts visual features layer-by-layer and generates the dynamic LoRA adapter.\n",
     "    \"\"\"\n",
+    "    # Ensure any patched LoRA hooks are cleared before feature extraction\n",
+    "    model.reset()\n",
+    "\n",
     "    internalize_messages = [\n",
     "        [\n",
     "            {\n",
@@ -571,6 +574,9 @@
     "    Runs inference comparing base model (using visual tokens) vs Video2LoRA (zero visual tokens).\n",
     "    \"\"\"\n",
     "    # === Part A: Base Model Inference (with visual tokens) ===\n",
+    "    # Ensure any patched LoRA hooks are cleared for a clean base VLM forward pass\n",
+    "    model.reset()\n",
+    "\n",
     "    base_messages = [\n",
     "        [\n",
     "            {\n",