Skip to content

Commit 401fb39

Browse files
Mary Hippmaryhipp
authored andcommitted
add FLUX control loras to starter models
1 parent 594511c commit 401fb39

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

invokeai/backend/model_manager/starter_models.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -488,6 +488,22 @@ class StarterModelBundles(BaseModel):
488488
type=ModelType.ControlNet,
489489
)
490490
# endregion
491+
# region Control LoRA
492+
flux_canny_control_lora = StarterModel(
493+
name="Hard Edge Detection (Canny)",
494+
base=BaseModelType.Flux,
495+
source="black-forest-labs/FLUX.1-Canny-dev-lora::flux1-canny-dev-lora.safetensors",
496+
description="Uses detected edges in the image to control composition.",
497+
type=ModelType.ControlLoRa,
498+
)
499+
flux_depth_control_lora = StarterModel(
500+
name="Depth Map",
501+
base=BaseModelType.Flux,
502+
source="black-forest-labs/FLUX.1-Depth-dev-lora::flux1-depth-dev-lora.safetensors",
503+
description="Uses depth information in the image to control the depth in the generation.",
504+
type=ModelType.ControlLoRa,
505+
)
506+
# endregion
491507
# region T2I Adapter
492508
t2i_canny_sd1 = StarterModel(
493509
name="Hard Edge Detection (canny)",
@@ -630,6 +646,8 @@ class StarterModelBundles(BaseModel):
630646
tile_sdxl,
631647
union_cnet_sdxl,
632648
union_cnet_flux,
649+
flux_canny_control_lora,
650+
flux_depth_control_lora,
633651
t2i_canny_sd1,
634652
t2i_sketch_sd1,
635653
t2i_depth_sd1,
@@ -688,6 +706,8 @@ class StarterModelBundles(BaseModel):
688706
clip_l_encoder,
689707
union_cnet_flux,
690708
ip_adapter_flux,
709+
flux_canny_control_lora,
710+
flux_depth_control_lora,
691711
]
692712

693713
STARTER_BUNDLES: dict[str, list[StarterModel]] = {

0 commit comments

Comments
 (0)