mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: added a check for base model

Browse Source 
added a check to make sure that the downloaded model is SDXL
This commit is contained in:
uptotec 2024-02-05 13:57:48 +02:00
parent 7fb4a701f3
commit 0895a6ae44
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fooocus_colab.ipynb
View File

@ -87,6 +87,8 @@
"if selected_version is None:\n", "if selected_version is None:\n",
" raise RuntimeError(\"this version doesn't exist\")\n", " raise RuntimeError(\"this version doesn't exist\")\n",
"\n", "\n",
"if \"SDXL\" not in selected_version.get('baseModel'):\n",
" raise RuntimeError(\"this model is not SDXL\")\n",
"\n", "\n",
"files = selected_version.get('files')\n", "files = selected_version.get('files')\n",
"primary_file = None\n", "primary_file = None\n",