github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-14 13:44:59 -05:00
Code Issues Packages Projects Releases Wiki Activity

Use the Shakker-Labs ControlNet union model as the only FLUX ControlNet starter model.

Browse Source
This commit is contained in:
Ryan Dick
2024-10-10 13:59:59 +00:00
parent 7f7d8e5177
commit bb6d073828
1 changed files with 2 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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