ViTeX-Edit-14B / diffsynth /utils /state_dict_converters /wan_video_image_encoder.py
ViTeX-Bench's picture
Bundle diffsynth library (no external repo dependency)
bc8c4af verified
def WanImageEncoderStateDictConverter(state_dict):
state_dict_ = {}
for name in state_dict:
if name.startswith("textual."):
continue
name_ = "model." + name
state_dict_[name_] = state_dict[name]
return state_dict_