Add conversion method to config in gpt-oss architecture adapter

degenfabian · degenfabian · commit 6023f6e9380b · 2025-08-14T16:59:51.000+02:00
diff --git a/transformer_lens/model_bridge/supported_architectures/gpt_oss.py b/transformer_lens/model_bridge/supported_architectures/gpt_oss.py
@@ -47,7 +47,12 @@ def __init__(self, cfg: Any) -> None:
                             "experts": BlockBridge(
                                 name="experts",
                                 submodules={
-                                    "gate_up": JointGateUpMLPBridge(name="gate_up_proj"),
+                                    "gate_up": JointGateUpMLPBridge(
+                                        name="gate_up_proj",
+                                        gate_up_config={
+                                            "split_gate_up_matrix": self.split_gate_up_matrix
+                                        },
+                                    ),
                                     "down": LinearBridge(name="down_proj"),
                                 },
                             ),
