From 65a2bc20bb924d8dff391fe3dd55830b0c9721aa Mon Sep 17 00:00:00 2001
From: City <125218114+city96@users.noreply.github.com>
Date: Fri, 28 Mar 2025 04:21:26 +0100
Subject: [PATCH] wan bf16 fix

---
 tools/convert.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tools/convert.py b/tools/convert.py
index 80ff3fd..52d1800 100644
--- a/tools/convert.py
+++ b/tools/convert.py
@@ -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"