theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-23 00:58:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

flex full support

Browse Source
This commit is contained in:
Artiprocher
2025-04-25 12:23:29 +08:00
parent 419ace37f3
commit cc6306136c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/models/flux_dit.py
View File

@@ -740,7 +740,7 @@ class FluxDiTStateDictConverter:
pass pass
if "guidance_embedder.timestep_embedder.0.weight" not in state_dict_: if "guidance_embedder.timestep_embedder.0.weight" not in state_dict_:
return state_dict_, {"disable_guidance_embedder": True} 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} return state_dict_, {"input_dim": 196, "num_blocks": 8}
else: else:
return state_dict_ return state_dict_