Upload 3 files

Invoke/INITIAL_MODELS.yaml

@@ -1,11 +1,4 @@
 # This file predefines a few models that the user may want to install.
-
-sd-1/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors:
-   description: hilloutmix_NiPrunedFp32Fix-inpainting  (4.27 GB)
-   repo_id: mnvvc/abcabc
-   recommended: True
-   default: True
-
 sd-1/main/stable-diffusion-v1-5:
    description: Stable Diffusion version 1.5 diffusers model (4.27 GB)
    repo_id: runwayml/stable-diffusion-v1-5
@@ -73,7 +66,7 @@ sd-1/main/waifu-diffusion:
    recommended: False
 sd-1/controlnet/canny:
    repo_id: lllyasviel/control_v11p_sd15_canny
-   recommended: True
+   recommended: False
 sd-1/controlnet/inpaint:
    repo_id: lllyasviel/control_v11p_sd15_inpaint
    recommended: False
@@ -82,7 +75,7 @@ sd-1/controlnet/mlsd:
    recommended: False
 sd-1/controlnet/depth:
    repo_id: lllyasviel/control_v11f1p_sd15_depth
-   recommended: True
+   recommended: False
 sd-1/controlnet/normal_bae:
    repo_id: lllyasviel/control_v11p_sd15_normalbae
    recommended: False

Invoke/_main.ipynb

@@ -32,9 +32,10 @@
     "# 模型下载\n",
     "!wget https://huggingface.co/Vanwise/sd-colab/resolve/main/Invoke/INITIAL_MODELS.yaml -O /content/IAekovnI/invokeai/configs/INITIAL_MODELS.yaml\n",
     "!python /content/IAekovnI/scripts/invokeai-model-install.py --root_dir /content/IAekovnI/db --yes\n",
+    "!curl -Lo /content/IAekovnI/db/models/sd-1/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors https://huggingface.co/spaces/weo1101/111/resolve/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors\n",
     "\n",
     "!wget https://huggingface.co/Vanwise/sd-colab/resolve/main/libtcmalloc_minimal.so.4\n",
-    "os.environ['LD_PRELOAD'] = '/content/libtcmalloc_minimal.so.4'\n",
+    "os.environ['LD_PRELOAD'] = '/content/IAekovnI/libtcmalloc_minimal.so.4'\n",
     "\n",
     "!wget https://huggingface.co/Vanwise/sd-colab/resolve/main/Invoke/_start.ipynb\n",
     "\n",

Invoke/_start.ipynb

@@ -7,6 +7,25 @@
    "outputs": [],
    "source": [
     "import threading\n",
+    "import yaml\n",
+    "\n",
+    "with open(\"/content/IAekovnI/db/configs/models.yaml\", \"r\") as f:\n",
+    "    models = yaml.safe_load(f)\n",
+    "\n",
+    "new_models = {\n",
+    "    \"sd-1/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting\": {\n",
+    "        \"path\": \"/content/IAekovnI/db/models/sd-1/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors\",\n",
+    "        \"config\": \"/content/IAekovnI/db/configs/stable-diffusion/v1-inpainting-inference.yaml\",\n",
+    "        \"variant\": \"inpaint\",\n",
+    "        \"format\": \"checkpoint\",\n",
+    "    },\n",
+    "}\n",
+    "\n",
+    "if \"sd-1/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting\" not in models:\n",
+    "    models.update(new_models)\n",
+    "\n",
+    "with open(\"/content/IAekovnI/db/configs/models.yaml\", \"w\") as f:\n",
+    "    yaml.safe_dump(models, f)\n",
     "\n",
     "def start_tunnle():\n",
     "    if use_ngrok:\n",