diff --git a/diffsynth/configs/model_config.py b/diffsynth/configs/model_config.py index 1a43700..aa737f7 100644 --- a/diffsynth/configs/model_config.py +++ b/diffsynth/configs/model_config.py @@ -85,6 +85,7 @@ model_loader_configs = [ (None, "21ea55f476dfc4fd135587abb59dfe5d", ["flux_vae_encoder", "flux_vae_decoder"], [FluxVAEEncoder, FluxVAEDecoder], "civitai"), (None, "a29710fea6dddb0314663ee823598e50", ["flux_dit"], [FluxDiT], "civitai"), (None, "57b02550baab820169365b3ee3afa2c9", ["flux_dit"], [FluxDiT], "civitai"), + (None, "3394f306c4cbf04334b712bf5aaed95f", ["flux_dit"], [FluxDiT], "civitai"), (None, "280189ee084bca10f70907bf6ce1649d", ["cog_vae_encoder", "cog_vae_decoder"], [CogVAEEncoder, CogVAEDecoder], "diffusers"), (None, "9b9313d104ac4df27991352fec013fd4", ["rife"], [IFNet], "civitai"), (None, "6b7116078c4170bfbeaedc8fe71f6649", ["esrgan"], [RRDBNet], "civitai"), diff --git a/diffsynth/models/flux_dit.py b/diffsynth/models/flux_dit.py index faf58cd..6011495 100644 --- a/diffsynth/models/flux_dit.py +++ b/diffsynth/models/flux_dit.py @@ -648,6 +648,8 @@ class FluxDiTStateDictConverter: } state_dict_ = {} for name, param in state_dict.items(): + if name.startswith("model.diffusion_model."): + name = name[len("model.diffusion_model."):] names = name.split(".") if name in rename_dict: rename = rename_dict[name]