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"