diff --git a/notebooks/263-latent-consistency-models-image-generation/263-latent-consistency-models-image-generation.ipynb b/notebooks/263-latent-consistency-models-image-generation/263-latent-consistency-models-image-generation.ipynb
index 849613cfc31..1c77513e779 100644
--- a/notebooks/263-latent-consistency-models-image-generation/263-latent-consistency-models-image-generation.ipynb
+++ b/notebooks/263-latent-consistency-models-image-generation/263-latent-consistency-models-image-generation.ipynb
@@ -929,6 +929,9 @@
     "import sys\n",
     "sys.path.append(\"../utils\")\n",
     "\n",
+    "if to_quantize.value and \"GPU\" in device.value:\n",
+    "    to_quantize.value = False\n",
+    "\n",
     "%load_ext skip_kernel_extension"
    ]
   },
@@ -950,9 +953,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "if to_quantize.value and \"GPU\" in device.value:\n",
-    "    to_quantize.value = False\n",
-    "\n",
     "%%skip not $to_quantize.value\n",
     "\n",
     "import datasets\n",