From 5f4cb3225552c6f873b64f012f795d1a0a5c107a Mon Sep 17 00:00:00 2001 From: "xuyixuan.xyx" Date: Tue, 22 Apr 2025 13:45:49 +0800 Subject: [PATCH] liblib-controlnet --- diffsynth/configs/model_config.py | 1 + diffsynth/models/flux_controlnet.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/diffsynth/configs/model_config.py b/diffsynth/configs/model_config.py index 4f540f9..c0bb673 100644 --- a/diffsynth/configs/model_config.py +++ b/diffsynth/configs/model_config.py @@ -108,6 +108,7 @@ model_loader_configs = [ (None, "52357cb26250681367488a8954c271e8", ["flux_controlnet"], [FluxControlNet], "diffusers"), (None, "0cfd1740758423a2a854d67c136d1e8c", ["flux_controlnet"], [FluxControlNet], "diffusers"), (None, "7f9583eb8ba86642abb9a21a4b2c9e16", ["flux_controlnet"], [FluxControlNet], "diffusers"), + (None, "43ad5aaa27dd4ee01b832ed16773fa52", ["flux_controlnet"], [FluxControlNet], "diffusers"), (None, "c07c0f04f5ff55e86b4e937c7a40d481", ["infiniteyou_image_projector"], [InfiniteYouImageProjector], "diffusers"), (None, "4daaa66cc656a8fe369908693dad0a35", ["flux_ipadapter"], [FluxIpAdapter], "diffusers"), (None, "51aed3d27d482fceb5e0739b03060e8f", ["sd3_dit", "sd3_vae_encoder", "sd3_vae_decoder"], [SD3DiT, SD3VAEEncoder, SD3VAEDecoder], "civitai"), diff --git a/diffsynth/models/flux_controlnet.py b/diffsynth/models/flux_controlnet.py index 7bc3dc0..85fccd7 100644 --- a/diffsynth/models/flux_controlnet.py +++ b/diffsynth/models/flux_controlnet.py @@ -320,6 +320,8 @@ class FluxControlNetStateDictConverter: extra_kwargs = {"num_joint_blocks": 4, "num_single_blocks": 1} elif hash_value == "7f9583eb8ba86642abb9a21a4b2c9e16": extra_kwargs = {"num_joint_blocks": 4, "num_single_blocks": 10} + elif hash_value == "43ad5aaa27dd4ee01b832ed16773fa52": + extra_kwargs = {"num_joint_blocks": 6, "num_single_blocks": 0} else: extra_kwargs = {} return state_dict_, extra_kwargs