Files
DiffSynth-Studio/diffsynth/utils/state_dict_converters/wan_video_animate_adapter.py
2025-12-04 16:33:07 +08:00

6 lines
333 B
Python

def WanAnimateAdapterStateDictConverter(state_dict):
state_dict_ = {}
for name in state_dict:
if name.startswith("pose_patch_embedding.") or name.startswith("face_adapter") or name.startswith("face_encoder") or name.startswith("motion_encoder"):
state_dict_[name] = state_dict[name]
return state_dict_