Fix remaining gate_proj/up_proj -> w1/w3 references in paired_names

2026-05-07 00:05:55 +00:00
parent 771e42cef3
commit c40607053b
1 changed files with 2 additions and 2 deletions
--- a/fp8_to_nvfp4_streaming.py
+++ b/fp8_to_nvfp4_streaming.py
@@ -293,8 +293,8 @@ def build_plan(src_dir: Path):
    for base, parts in expert_pairs.items():
        if "w1" in parts and "w3" in parts:
            expert_pair_groups.append((base, parts["w1"], parts["w3"]))
-            paired_names.add(parts["gate_proj"])
-            paired_names.add(parts["up_proj"])
+            paired_names.add(parts["w1"])
+            paired_names.add(parts["w3"])

    # Classify everything else
    solo_quantize = []