From 102b47190fe2ed50694e4b89824ad8a5ad41a273 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Fri, 26 Jul 2024 08:08:53 +1000 Subject: [PATCH] feat(ui): update qr code cnet starter model - For SD1.5, use the new V2 version - Add the SDXL version --- .../backend/model_manager/starter_models.py | 73 ++++++++++--------- 1 file changed, 40 insertions(+), 33 deletions(-) diff --git a/invokeai/backend/model_manager/starter_models.py b/invokeai/backend/model_manager/starter_models.py index e1526392f6..c460a5e86e 100644 --- a/invokeai/backend/model_manager/starter_models.py +++ b/invokeai/backend/model_manager/starter_models.py @@ -187,164 +187,171 @@ STARTER_MODELS: list[StarterModel] = [ # endregion # region ControlNet StarterModel( - name="QRCode Monster", + name="QRCode Monster v2 (SD1.5)", base=BaseModelType.StableDiffusion1, - source="monster-labs/control_v1p_sd15_qrcode_monster", - description="Controlnet model that generates scannable creative QR codes", + source="monster-labs/control_v1p_sd15_qrcode_monster::v2", + 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, ), StarterModel( name="canny", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="inpaint", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="mlsd", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="depth", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="normal_bae", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="seg", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="lineart", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="lineart_anime", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="openpose", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="scribble", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="softedge", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="shuffle", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="tile", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="ip2p", base=BaseModelType.StableDiffusion1, 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, ), StarterModel( name="canny-sdxl", base=BaseModelType.StableDiffusionXL, - source="xinsir/controlnet-canny-sdxl-1.0", - description="Controlnet weights trained on sdxl-1.0 with canny conditioning, by Xinsir.", + source="xinsir/controlNet-canny-sdxl-1.0", + description="ControlNet weights trained on sdxl-1.0 with canny conditioning, by Xinsir.", type=ModelType.ControlNet, ), StarterModel( name="depth-sdxl", base=BaseModelType.StableDiffusionXL, - source="diffusers/controlnet-depth-sdxl-1.0", - description="Controlnet weights trained on sdxl-1.0 with depth conditioning.", + source="diffusers/controlNet-depth-sdxl-1.0", + description="ControlNet weights trained on sdxl-1.0 with depth conditioning.", type=ModelType.ControlNet, ), StarterModel( name="softedge-dexined-sdxl", base=BaseModelType.StableDiffusionXL, - source="SargeZT/controlnet-sd-xl-1.0-softedge-dexined", - description="Controlnet weights trained on sdxl-1.0 with dexined soft edge preprocessing.", + source="SargeZT/controlNet-sd-xl-1.0-softedge-dexined", + description="ControlNet weights trained on sdxl-1.0 with dexined soft edge preprocessing.", type=ModelType.ControlNet, ), StarterModel( name="depth-16bit-zoe-sdxl", base=BaseModelType.StableDiffusionXL, - source="SargeZT/controlnet-sd-xl-1.0-depth-16bit-zoe", - description="Controlnet weights trained on sdxl-1.0 with Zoe's preprocessor (16 bits).", + source="SargeZT/controlNet-sd-xl-1.0-depth-16bit-zoe", + description="ControlNet weights trained on sdxl-1.0 with Zoe's preprocessor (16 bits).", type=ModelType.ControlNet, ), StarterModel( name="depth-zoe-sdxl", base=BaseModelType.StableDiffusionXL, - source="diffusers/controlnet-zoe-depth-sdxl-1.0", - description="Controlnet weights trained on sdxl-1.0 with Zoe's preprocessor (32 bits).", + source="diffusers/controlNet-zoe-depth-sdxl-1.0", + description="ControlNet weights trained on sdxl-1.0 with Zoe's preprocessor (32 bits).", type=ModelType.ControlNet, ), StarterModel( name="openpose-sdxl", base=BaseModelType.StableDiffusionXL, - source="xinsir/controlnet-openpose-sdxl-1.0", - description="Controlnet weights trained on sdxl-1.0 compatible with the DWPose processor by Xinsir.", + source="xinsir/controlNet-openpose-sdxl-1.0", + description="ControlNet weights trained on sdxl-1.0 compatible with the DWPose processor by Xinsir.", type=ModelType.ControlNet, ), StarterModel( name="scribble-sdxl", base=BaseModelType.StableDiffusionXL, - 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.", + 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.", type=ModelType.ControlNet, ), StarterModel( name="tile-sdxl", base=BaseModelType.StableDiffusionXL, - source="xinsir/controlnet-tile-sdxl-1.0", - description="Controlnet weights trained on sdxl-1.0 with tiled image conditioning", + source="xinsir/controlNet-tile-sdxl-1.0", + description="ControlNet weights trained on sdxl-1.0 with tiled image conditioning", type=ModelType.ControlNet, ), # endregion