From bb6d073828c9d11072815c82e5869249e79f79ce Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Thu, 10 Oct 2024 13:59:59 +0000 Subject: [PATCH] Use the Shakker-Labs ControlNet union model as the only FLUX ControlNet starter model. --- .../backend/model_manager/starter_models.py | 32 ++----------------- 1 file changed, 2 insertions(+), 30 deletions(-) diff --git a/invokeai/backend/model_manager/starter_models.py b/invokeai/backend/model_manager/starter_models.py index 8b83d4c577..093768f399 100644 --- a/invokeai/backend/model_manager/starter_models.py +++ b/invokeai/backend/model_manager/starter_models.py @@ -423,37 +423,9 @@ STARTER_MODELS: list[StarterModel] = [ type=ModelType.ControlNet, ), StarterModel( - name="flux-canny-controlnet-v3", + name="FLUX.1-dev-Controlnet-Union-Pro", base=BaseModelType.Flux, - source="https://huggingface.co/XLabs-AI/flux-controlnet-collections/resolve/main/flux-canny-controlnet-v3.safetensors", - description="ControlNet weights trained on FLUX with canny image conditioning", - type=ModelType.ControlNet, - ), - StarterModel( - name="flux-depth-controlnet-v3", - base=BaseModelType.Flux, - source="https://huggingface.co/XLabs-AI/flux-controlnet-collections/resolve/main/flux-depth-controlnet-v3.safetensors", - description="ControlNet weights trained on FLUX with depth image conditioning", - type=ModelType.ControlNet, - ), - StarterModel( - name="flux-hed-controlnet-v3", - base=BaseModelType.Flux, - source="https://huggingface.co/XLabs-AI/flux-controlnet-collections/resolve/main/flux-hed-controlnet-v3.safetensors", - description="ControlNet weights trained on FLUX with HED image conditioning", - type=ModelType.ControlNet, - ), - StarterModel( - name="FLUX.1-dev-Controlnet-Canny", - base=BaseModelType.Flux, - source="InstantX/FLUX.1-dev-Controlnet-Canny", - description="ControlNet weights trained on FLUX with Canny image conditioning", - type=ModelType.ControlNet, - ), - StarterModel( - name="FLUX.1-dev-Controlnet-Union", - base=BaseModelType.Flux, - source="InstantX/FLUX.1-dev-Controlnet-Union", + source="Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro", description="A unified ControlNet for FLUX.1-dev model that supports 7 control modes, including canny (0), tile (1), depth (2), blur (3), pose (4), gray (5), low quality (6)", type=ModelType.ControlNet, ),