fix skip quantization in lcm (#1427)

openvinotoolkit · Nov 6, 2023 · 9da8e5e · 9da8e5e
1 parent 3cebaab
commit 9da8e5e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...-consistency-models-image-generation/263-latent-consistency-models-image-generation.ipynb b/...-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",