diff --git a/examples/sam3_agent.ipynb b/examples/sam3_agent.ipynb
index 23177e436..81d4064ad 100644
--- a/examples/sam3_agent.ipynb
+++ b/examples/sam3_agent.ipynb
@@ -53,7 +53,7 @@
         "torch.autocast(\"cuda\", dtype=torch.bfloat16).__enter__()\n",
         "\n",
         "# inference mode for the whole notebook. Disable if you need gradients\n",
-        "torch.inference_mode().__enter__()"
+        "torch.set_grad_enabled(False)"
       ]
     },
     {
diff --git a/examples/sam3_image_batched_inference.ipynb b/examples/sam3_image_batched_inference.ipynb
index dddbfd839..8cd123646 100644
--- a/examples/sam3_image_batched_inference.ipynb
+++ b/examples/sam3_image_batched_inference.ipynb
@@ -81,7 +81,7 @@
     "torch.autocast(\"cuda\", dtype=torch.bfloat16).__enter__()\n",
     "\n",
     "# inference mode for the whole notebook. Disable if you need gradients\n",
-    "torch.inference_mode().__enter__()\n"
+    "torch.set_grad_enabled(False)\n"
    ]
   },
   {
diff --git a/examples/sam3_image_interactive.ipynb b/examples/sam3_image_interactive.ipynb
index 7e1e76663..02a19c95d 100644
--- a/examples/sam3_image_interactive.ipynb
+++ b/examples/sam3_image_interactive.ipynb
@@ -75,7 +75,7 @@
     "torch.autocast(\"cuda\", dtype=torch.bfloat16).__enter__()\n",
     "\n",
     "# inference mode for the whole notebook. Disable if you need gradients\n",
-    "torch.inference_mode().__enter__()"
+    "torch.set_grad_enabled(False)"
    ]
   },
   {