From 0895a6ae44b657dc9e370a3f51efef02d2b4f28d Mon Sep 17 00:00:00 2001 From: uptotec Date: Mon, 5 Feb 2024 13:57:48 +0200 Subject: [PATCH] feat: added a check for base model added a check to make sure that the downloaded model is SDXL --- fooocus_colab.ipynb | 2 ++ 1 file changed, 2 insertions(+) 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",