Support mixed precision fp8 scaled models

2025-12-22 21:40:14 +02:00
parent 41683a0423
commit 3e45021422
1 changed files with 2 additions and 0 deletions
@@ -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...")