Support MERJIC's new model (#298)

* Update flux_dit.py
* Update model_config.py
This commit is contained in:
Zhongjie Duan
2024-12-28 21:21:25 +08:00
committed by GitHub
parent 6cc5fd6d1e
commit 3228c3e085
2 changed files with 3 additions and 0 deletions

View File

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