github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-01-14 20:38:08 -05:00
Code Issues Packages Projects Releases Wiki Activity

(api) make sure all controlnet starter models will still have pre-processors correctly assigned when probed based on name

Browse Source
This commit is contained in:
maryhipp
2024-10-23 13:49:35 -04:00
committed by psychedelicious
parent afa5a4b17c
commit 43cc4684e1
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/backend/model_manager/probe.py
View File

@@ -462,8 +462,9 @@ MODEL_NAME_TO_PREPROCESSOR = {
"normal": "normalbae_image_processor",
"sketch": "pidi_image_processor",
"scribble": "lineart_image_processor",
"lineart": "lineart_image_processor",
"lineart anime": "lineart_anime_image_processor",
"lineart_anime": "lineart_anime_image_processor",
"lineart": "lineart_image_processor",
"softedge": "hed_image_processor",
"hed": "hed_image_processor",
"shuffle": "content_shuffle_image_processor",

8
invokeai/backend/model_manager/starter_models.py
View File

@@ -323,7 +323,7 @@ inpaint_cnet_sd1 = StarterModel(
previous_names=["inpaint"],
)
mlsd_sd1 = StarterModel(
name="Line Drawing",
name="Line Drawing (mlsd)",
base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11p_sd15_mlsd",
description="Uses straight line detection for controlling the generation.",
@@ -387,7 +387,7 @@ scribble_sd1 = StarterModel(
previous_names=["scribble"],
)
softedge_sd1 = StarterModel(
name="Soft Edge Detection",
name="Soft Edge Detection (softedge)",
base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11p_sd15_softedge",
description="Uses a soft edge detection map to control composition.",
@@ -395,7 +395,7 @@ softedge_sd1 = StarterModel(
previous_names=["softedge"],
)
shuffle_sd1 = StarterModel(
name="Remix",
name="Remix (shuffle)",
base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11e_sd15_shuffle",
description="ControlNet weights trained on sd-1.5 with shuffle image conditioning",
@@ -427,7 +427,7 @@ depth_sdxl = StarterModel(
previous_names=["depth-sdxl"],
)
softedge_sdxl = StarterModel(
name="Soft Edge Detection",
name="Soft Edge Detection (softedge)",
base=BaseModelType.StableDiffusionXL,
source="SargeZT/controlNet-sd-xl-1.0-softedge-dexined",
description="Uses a soft edge detection map to control composition.",