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...")