From 771e42cef3882e0e5ff89e143426be324cfa4412 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Thu, 7 May 2026 00:05:25 +0000 Subject: [PATCH] Fix expert pair dict keys: w1/w3 not gate_proj/up_proj --- 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 186fd5f..5106cfc 100644 --- a/fp8_to_nvfp4_streaming.py +++ b/fp8_to_nvfp4_streaming.py @@ -291,8 +291,8 @@ def build_plan(src_dir: Path): paired_names = set() expert_pair_groups = [] for base, parts in expert_pairs.items(): - if "gate_proj" in parts and "up_proj" in parts: - expert_pair_groups.append((base, parts["gate_proj"], parts["up_proj"])) + 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"])