qwen-video-edit / diffsynth /utils /state_dict_converters /stable_diffusion_text_encoder.py
multimodalart's picture
multimodalart HF Staff
Upload folder using huggingface_hub
b347817 verified
Raw
History Blame Contribute Delete
289 Bytes
def SDTextEncoderStateDictConverter(state_dict):
new_state_dict = {}
for key in state_dict:
if key.startswith("text_model.") and "position_ids" not in key:
new_key = "model." + key
new_state_dict[new_key] = state_dict[key]
return new_state_dict