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, ),