From 3e450214225c2934d269f2398378f0af477fd239 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 22 Dec 2025 21:40:14 +0200
Subject: [PATCH] Support mixed precision fp8 scaled models

---
 nodes_model_loading.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/nodes_model_loading.py b/nodes_model_loading.py
index d07a71a..e2c24d7 100644
--- a/nodes_model_loading.py
+++ b/nodes_model_loading.py
@@ -1480,6 +1480,8 @@ class WanVideoModelLoader:
             sd.update(extra_sd)
             del extra_sd
 
+        sd = {k.replace(".weight_scale", ".scale_weight"): v for k, v in sd.items()}
+
         # FlashVSR
         if "LQ_proj_in.norm1.gamma" in sd:
             log.info("FlashVSR model detected, patching model...")