support wan-series models

2026-03-25 02:38:10 +00:00 · 2025-11-13 17:30:19 +08:00
parent cb70126c88
commit 5be5c32fe4
64 changed files with 9915 additions and 24 deletions
--- a/diffsynth/utils/state_dict_converters/wan_video_animate_adapter.py
+++ b/diffsynth/utils/state_dict_converters/wan_video_animate_adapter.py
@@ -0,0 +1,6 @@
+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_