wan bf16 fix

2025-03-28 04:21:26 +01:00
parent 1c873edeb6
commit 65a2bc20bb
1 changed files with 3 additions and 0 deletions
@@ -73,6 +73,9 @@ class ModelWan(ModelHyVid):
            "head.modulation",
        )
    ]
+    keys_hiprec = [
+        ".modulation" # nn.parameter, can't load from BF16 ver
+    ]

 class ModelLTXV(ModelTemplate):
    arch = "ltxv"