@@ -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
492508t2i_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
693713STARTER_BUNDLES : dict [str , list [StarterModel ]] = {
0 commit comments