From cc6306136cd129ee04635edf40cbbe730d5bce86 Mon Sep 17 00:00:00 2001 From: Artiprocher Date: Fri, 25 Apr 2025 12:23:29 +0800 Subject: [PATCH] flex full support --- diffsynth/models/flux_dit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/diffsynth/models/flux_dit.py b/diffsynth/models/flux_dit.py index 31dde9c..ea5ce21 100644 --- a/diffsynth/models/flux_dit.py +++ b/diffsynth/models/flux_dit.py @@ -740,7 +740,7 @@ class FluxDiTStateDictConverter: pass if "guidance_embedder.timestep_embedder.0.weight" not in state_dict_: return state_dict_, {"disable_guidance_embedder": True} - elif "double_blocks.8.img_attn.norm.key_norm.scale" not in state_dict_: + elif "blocks.8.attn.norm_k_a.weight" not in state_dict_: return state_dict_, {"input_dim": 196, "num_blocks": 8} else: return state_dict_