From 35eb0763ab4cd195058b06e935fc14480a1791b9 Mon Sep 17 00:00:00 2001 From: Haoming Date: Fri, 3 Apr 2026 23:36:39 +0800 Subject: [PATCH] detection --- modules_forge/packages/huggingface_guess/detection.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules_forge/packages/huggingface_guess/detection.py b/modules_forge/packages/huggingface_guess/detection.py index 87115606..93f498f9 100644 --- a/modules_forge/packages/huggingface_guess/detection.py +++ b/modules_forge/packages/huggingface_guess/detection.py @@ -128,7 +128,7 @@ def detect_unet_config(state_dict: dict, key_prefix: str) -> dict: dit_config["guidance_embed"] = True return dit_config - if "{}double_blocks.0.img_attn.norm.key_norm.scale".format(key_prefix) in state_dict_keys and ("{}img_in.weight".format(key_prefix) in state_dict_keys or f"{key_prefix}distilled_guidance_layer.norms.0.scale" in state_dict_keys): # Flux.1 / Flux.2 + if ("{}double_blocks.0.img_attn.norm.key_norm.scale".format(key_prefix) in state_dict_keys or "{}double_blocks.0.img_attn.norm.key_norm.weight".format(key_prefix) in state_dict_keys) and ("{}img_in.weight".format(key_prefix) in state_dict_keys or f"{key_prefix}distilled_guidance_layer.norms.0.scale" in state_dict_keys): # Flux.1 / Flux.2 dit_config = {} if "{}double_stream_modulation_img.lin.weight".format(key_prefix) in state_dict_keys: dit_config["image_model"] = "flux2"