CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

detection

Browse Source
This commit is contained in:
Haoming 2026-04-03 23:36:39 +08:00
parent 2d1f34f9c7
commit 35eb0763ab
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules_forge/packages/huggingface_guess/detection.py
View File

@ -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"