support wan2.2 5B T2V

2026-03-23 09:28:12 +00:00 · 2025-07-20 18:13:50 +08:00
parent d19fcc8c04
commit 55951590f5
5 changed files with 628 additions and 3 deletions
--- a/diffsynth/models/wan_video_dit.py
+++ b/diffsynth/models/wan_video_dit.py
@@ -659,6 +659,20 @@ class WanModelStateDictConverter:
                "add_control_adapter": True,
                "in_dim_control_adapter": 24,
            }
+        elif hash_state_dict_keys(state_dict) == "1f5ab7703c6fc803fdded85ff040c316":
+            config = {
+                "has_image_input": False,
+                "patch_size": [1, 2, 2],
+                "in_dim": 48,
+                "dim": 3072,
+                "ffn_dim": 14336,
+                "freq_dim": 256,
+                "text_dim": 4096,
+                "out_dim": 48,
+                "num_heads": 24,
+                "num_layers": 30,
+                "eps": 1e-6,
+            }
        else:
            config = {}
        return state_dict, config