diffsynth/utils/state_dict_converters/z_image_dit.py

def ZImageDiTStateDictConverter(state_dict):
    state_dict_ = {name.replace("model.diffusion_model.", ""): state_dict[name] for name in state_dict}
    return state_dict_