diff --git a/nodes_model_loading.py b/nodes_model_loading.py index 29e022d..dedd060 100644 --- a/nodes_model_loading.py +++ b/nodes_model_loading.py @@ -1128,7 +1128,7 @@ class WanVideoModelLoader: scale_weights = {} if "fp8" in quantization: for k, v in sd.items(): - if k.endswith(".scale_weight"): + if k.endswith(".scale_weight") or k.endswith(".weight_scale"): is_scaled_fp8 = True break