From c40607053b3d5ca156c9d233ff8159add02ea892 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Thu, 7 May 2026 00:05:55 +0000 Subject: [PATCH] Fix remaining gate_proj/up_proj -> w1/w3 references in paired_names --- fp8_to_nvfp4_streaming.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fp8_to_nvfp4_streaming.py b/fp8_to_nvfp4_streaming.py index 5106cfc..bdbcf08 100644 --- 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 = []