Function rename_state_dict

diffusers/scripts/convert_consistency_decoder.py:503–506

(sd, embedding)

def rename_state_dict(sd, embedding):
    sd = {rename_state_dict_key(k): v for k, v in sd.items()}
    sd["embed_time.emb.weight"] = embedding["weight"]
    return sd


# encode with stable diffusion vae