diff --git a/fooocus_colab.ipynb b/fooocus_colab.ipynb
index daa99e0..b3a1c80 100644
--- a/fooocus_colab.ipynb
+++ b/fooocus_colab.ipynb
@@ -87,6 +87,8 @@
     "if selected_version is None:\n",
     "  raise RuntimeError(\"this version doesn't exist\")\n",
     "\n",
+    "if \"SDXL\" not in selected_version.get('baseModel'):\n",
+    "  raise RuntimeError(\"this model is not SDXL\")\n",
     "\n",
     "files = selected_version.get('files')\n",
     "primary_file = None\n",