Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(ui): update qr code cnet starter model

Browse Source 
- For SD1.5, use the new V2 version
- Add the SDXL version
This commit is contained in:
psychedelicious 2024-07-26 08:08:53 +10:00
parent 269fe2e3bb
commit 102b47190f
1 changed files with 40 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -187,164 +187,171 @@ STARTER_MODELS: list[StarterModel] = [
# endregion # endregion
# region ControlNet # region ControlNet
StarterModel( StarterModel(
name="QRCode Monster", name="QRCode Monster v2 (SD1.5)",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="monster-labs/control_v1p_sd15_qrcode_monster", source="monster-labs/control_v1p_sd15_qrcode_monster::v2",
description="Controlnet model that generates scannable creative QR codes", description="ControlNet model that generates scannable creative QR codes",
type=ModelType.ControlNet,
),
StarterModel(
name="QRCode Monster (SDXL)",
base=BaseModelType.StableDiffusionXL,
source="monster-labs/control_v1p_sdxl_qrcode_monster",
description="ControlNet model that generates scannable creative QR codes",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="canny", name="canny",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11p_sd15_canny", source="lllyasviel/control_v11p_sd15_canny",
description="Controlnet weights trained on sd-1.5 with canny conditioning.", description="ControlNet weights trained on sd-1.5 with canny conditioning.",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="inpaint", name="inpaint",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11p_sd15_inpaint", source="lllyasviel/control_v11p_sd15_inpaint",
description="Controlnet weights trained on sd-1.5 with canny conditioning, inpaint version", description="ControlNet weights trained on sd-1.5 with canny conditioning, inpaint version",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="mlsd", name="mlsd",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11p_sd15_mlsd", source="lllyasviel/control_v11p_sd15_mlsd",
description="Controlnet weights trained on sd-1.5 with canny conditioning, MLSD version", description="ControlNet weights trained on sd-1.5 with canny conditioning, MLSD version",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="depth", name="depth",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11f1p_sd15_depth", source="lllyasviel/control_v11f1p_sd15_depth",
description="Controlnet weights trained on sd-1.5 with depth conditioning", description="ControlNet weights trained on sd-1.5 with depth conditioning",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="normal_bae", name="normal_bae",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11p_sd15_normalbae", source="lllyasviel/control_v11p_sd15_normalbae",
description="Controlnet weights trained on sd-1.5 with normalbae image conditioning", description="ControlNet weights trained on sd-1.5 with normalbae image conditioning",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="seg", name="seg",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11p_sd15_seg", source="lllyasviel/control_v11p_sd15_seg",
description="Controlnet weights trained on sd-1.5 with seg image conditioning", description="ControlNet weights trained on sd-1.5 with seg image conditioning",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="lineart", name="lineart",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11p_sd15_lineart", source="lllyasviel/control_v11p_sd15_lineart",
description="Controlnet weights trained on sd-1.5 with lineart image conditioning", description="ControlNet weights trained on sd-1.5 with lineart image conditioning",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="lineart_anime", name="lineart_anime",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11p_sd15s2_lineart_anime", source="lllyasviel/control_v11p_sd15s2_lineart_anime",
description="Controlnet weights trained on sd-1.5 with anime image conditioning", description="ControlNet weights trained on sd-1.5 with anime image conditioning",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="openpose", name="openpose",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11p_sd15_openpose", source="lllyasviel/control_v11p_sd15_openpose",
description="Controlnet weights trained on sd-1.5 with openpose image conditioning", description="ControlNet weights trained on sd-1.5 with openpose image conditioning",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="scribble", name="scribble",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11p_sd15_scribble", source="lllyasviel/control_v11p_sd15_scribble",
description="Controlnet weights trained on sd-1.5 with scribble image conditioning", description="ControlNet weights trained on sd-1.5 with scribble image conditioning",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="softedge", name="softedge",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11p_sd15_softedge", source="lllyasviel/control_v11p_sd15_softedge",
description="Controlnet weights trained on sd-1.5 with soft edge conditioning", description="ControlNet weights trained on sd-1.5 with soft edge conditioning",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="shuffle", name="shuffle",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11e_sd15_shuffle", source="lllyasviel/control_v11e_sd15_shuffle",
description="Controlnet weights trained on sd-1.5 with shuffle image conditioning", description="ControlNet weights trained on sd-1.5 with shuffle image conditioning",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="tile", name="tile",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11f1e_sd15_tile", source="lllyasviel/control_v11f1e_sd15_tile",
description="Controlnet weights trained on sd-1.5 with tiled image conditioning", description="ControlNet weights trained on sd-1.5 with tiled image conditioning",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="ip2p", name="ip2p",
base=BaseModelType.StableDiffusion1, base=BaseModelType.StableDiffusion1,
source="lllyasviel/control_v11e_sd15_ip2p", source="lllyasviel/control_v11e_sd15_ip2p",
description="Controlnet weights trained on sd-1.5 with ip2p conditioning.", description="ControlNet weights trained on sd-1.5 with ip2p conditioning.",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="canny-sdxl", name="canny-sdxl",
base=BaseModelType.StableDiffusionXL, base=BaseModelType.StableDiffusionXL,
source="xinsir/controlnet-canny-sdxl-1.0", source="xinsir/controlNet-canny-sdxl-1.0",
description="Controlnet weights trained on sdxl-1.0 with canny conditioning, by Xinsir.", description="ControlNet weights trained on sdxl-1.0 with canny conditioning, by Xinsir.",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="depth-sdxl", name="depth-sdxl",
base=BaseModelType.StableDiffusionXL, base=BaseModelType.StableDiffusionXL,
source="diffusers/controlnet-depth-sdxl-1.0", source="diffusers/controlNet-depth-sdxl-1.0",
description="Controlnet weights trained on sdxl-1.0 with depth conditioning.", description="ControlNet weights trained on sdxl-1.0 with depth conditioning.",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="softedge-dexined-sdxl", name="softedge-dexined-sdxl",
base=BaseModelType.StableDiffusionXL, base=BaseModelType.StableDiffusionXL,
source="SargeZT/controlnet-sd-xl-1.0-softedge-dexined", source="SargeZT/controlNet-sd-xl-1.0-softedge-dexined",
description="Controlnet weights trained on sdxl-1.0 with dexined soft edge preprocessing.", description="ControlNet weights trained on sdxl-1.0 with dexined soft edge preprocessing.",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="depth-16bit-zoe-sdxl", name="depth-16bit-zoe-sdxl",
base=BaseModelType.StableDiffusionXL, base=BaseModelType.StableDiffusionXL,
source="SargeZT/controlnet-sd-xl-1.0-depth-16bit-zoe", source="SargeZT/controlNet-sd-xl-1.0-depth-16bit-zoe",
description="Controlnet weights trained on sdxl-1.0 with Zoe's preprocessor (16 bits).", description="ControlNet weights trained on sdxl-1.0 with Zoe's preprocessor (16 bits).",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="depth-zoe-sdxl", name="depth-zoe-sdxl",
base=BaseModelType.StableDiffusionXL, base=BaseModelType.StableDiffusionXL,
source="diffusers/controlnet-zoe-depth-sdxl-1.0", source="diffusers/controlNet-zoe-depth-sdxl-1.0",
description="Controlnet weights trained on sdxl-1.0 with Zoe's preprocessor (32 bits).", description="ControlNet weights trained on sdxl-1.0 with Zoe's preprocessor (32 bits).",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="openpose-sdxl", name="openpose-sdxl",
base=BaseModelType.StableDiffusionXL, base=BaseModelType.StableDiffusionXL,
source="xinsir/controlnet-openpose-sdxl-1.0", source="xinsir/controlNet-openpose-sdxl-1.0",
description="Controlnet weights trained on sdxl-1.0 compatible with the DWPose processor by Xinsir.", description="ControlNet weights trained on sdxl-1.0 compatible with the DWPose processor by Xinsir.",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="scribble-sdxl", name="scribble-sdxl",
base=BaseModelType.StableDiffusionXL, base=BaseModelType.StableDiffusionXL,
source="xinsir/controlnet-scribble-sdxl-1.0", source="xinsir/controlNet-scribble-sdxl-1.0",
description="Controlnet weights trained on sdxl-1.0 compatible with various lineart processors and black/white sketches by Xinsir.", description="ControlNet weights trained on sdxl-1.0 compatible with various lineart processors and black/white sketches by Xinsir.",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
StarterModel( StarterModel(
name="tile-sdxl", name="tile-sdxl",
base=BaseModelType.StableDiffusionXL, base=BaseModelType.StableDiffusionXL,
source="xinsir/controlnet-tile-sdxl-1.0", source="xinsir/controlNet-tile-sdxl-1.0",
description="Controlnet weights trained on sdxl-1.0 with tiled image conditioning", description="ControlNet weights trained on sdxl-1.0 with tiled image conditioning",
type=ModelType.ControlNet, type=ModelType.ControlNet,
), ),
# endregion # endregion