diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp
index d57f63b17..ae6ee803e 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py
index b3813d3a6..cd780c4ff 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py
@@ -88,6 +88,8 @@ class Transducer(nn.Module):
             for i in range(512):
                 mean, std = statistic[i].strip().split(' ')
                 new_emb[i] = torch.normal(float(mean), float(std), size=50).squeeze()
+            new_emb = new_emb.transpose(1,0)
+            self.prompt = torch.nn.Parameter(new_emb) 
             
 
     def forward(