PencilFolder / diffsynth /utils /state_dict_converters /flux_text_encoder_t5.py
PencilHu's picture
Upload folder using huggingface_hub
1146a67 verified
Raw
History Blame Contribute Delete
210 Bytes
def FluxTextEncoderT5StateDictConverter(state_dict):
state_dict_ = {i: state_dict[i] for i in state_dict}
state_dict_["encoder.embed_tokens.weight"] = state_dict["shared.weight"]
return state_dict_