missing keys error

Author: prishajain1

src/maxdiffusion/models/wan/wan_utils.py

@@ -278,10 +278,6 @@ def load_base_wan_transformer(
       if "norm_added_q" in renamed_pt_key:
            renamed_pt_key = renamed_pt_key.replace("weight", "scale")
            renamed_pt_key = renamed_pt_key.replace("kernel", "scale")
-           
-           # 3. Force 'weight' -> 'scale'
-           renamed_pt_key = renamed_pt_key.replace("weight", "scale")
-           renamed_pt_key = renamed_pt_key.replace("kernel", "scale")
       renamed_pt_key = renamed_pt_key.replace("blocks_", "blocks.")
       renamed_pt_key = renamed_pt_key.replace(".scale_shift_table", ".adaln_scale_shift_table")
       renamed_pt_key = renamed_pt_key.replace("to_out_0", "proj_attn")