support more wan models

2026-03-18 22:08:13 +00:00 · 2025-04-08 19:22:53 +08:00
parent f6c6e3c640
commit 3cc9764bc9
2 changed files with 15 additions and 0 deletions
--- a/diffsynth/models/wan_video_dit.py
+++ b/diffsynth/models/wan_video_dit.py
@@ -535,6 +535,20 @@ class WanModelStateDictConverter:
                "num_layers": 30,
                "eps": 1e-6
            }
+        elif hash_state_dict_keys(state_dict) == "efa44cddf936c70abd0ea28b6cbe946c":
+            config = {
+                "has_image_input": True,
+                "patch_size": [1, 2, 2],
+                "in_dim": 48,
+                "dim": 5120,
+                "ffn_dim": 13824,
+                "freq_dim": 256,
+                "text_dim": 4096,
+                "out_dim": 16,
+                "num_heads": 40,
+                "num_layers": 40,
+                "eps": 1e-6
+            }
        else:
            config = {}
        return state_dict, config