From 2d38e905095a1e3eb1a0fa199aee340d435a606f Mon Sep 17 00:00:00 2001
From: Gnome Ann <>
Date: Wed, 20 Apr 2022 12:56:57 -0400
Subject: [PATCH] Remove lm_head.weight from maps/xglm.json

---
 maps/xglm.json | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/maps/xglm.json b/maps/xglm.json
index 65ab5e5e..581f825c 100644
--- a/maps/xglm.json
+++ b/maps/xglm.json
@@ -9,8 +9,7 @@
   "static_weights": {
     "model.embed_tokens.weight": {"mtj": {"module": "embedding_shard/~/linear", "param": "w", "transforms": ["no_transpose", "vocab_pad"]}},
     "model.layer_norm.weight": {"mtj": {"module": "projection_shard/~/replicated_layer_norm", "param": "scale"}},
-    "model.layer_norm.bias": {"mtj": {"module": "projection_shard/~/replicated_layer_norm", "param": "offset"}},
-    "lm_head.weight": {"mtj": {"module": "projection_shard/~/linear", "param": "w", "transforms": ["vocab_pad"]}}
+    "model.layer_norm.bias": {"mtj": {"module": "projection_shard/~/replicated_layer_norm", "param": "offset"}}
   },
   "layer_weights": {
     "model.layers.{layer}.self_attn.q_proj.weight": {"mtj": {"module": "layer_{layer}/~/linear", "param": "w"}},