Rename Neo bridges to be in line with new naming scheme (#1056)

Author: degenfabian

transformer_lens/model_bridge/supported_architectures/neo.py

@@ -71,18 +71,18 @@ def __init__(self, cfg: Any) -> None:
                         name="attn.attention",
                         config=self.cfg,
                         submodules={
-                            "W_Q": LinearBridge(name="q_proj"),
-                            "W_K": LinearBridge(name="k_proj"),
-                            "W_V": LinearBridge(name="v_proj"),
-                            "W_O": LinearBridge(name="out_proj"),
+                            "q": LinearBridge(name="q_proj"),
+                            "k": LinearBridge(name="k_proj"),
+                            "v": LinearBridge(name="v_proj"),
+                            "o": LinearBridge(name="out_proj"),
                         },
                     ),
                     "ln2": NormalizationBridge(name="ln_2", config=self.cfg),
                     "mlp": MLPBridge(
                         name="mlp",
                         submodules={
-                            "W_in": LinearBridge(name="c_fc"),
-                            "W_out": LinearBridge(name="c_proj"),
+                            "in": LinearBridge(name="c_fc"),
+                            "out": LinearBridge(name="c_proj"),
                         },
                     ),
                 },