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 164d78417..b1038da3a 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 45f46b93e..83f101dea 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py
@@ -89,7 +89,7 @@ class Transducer(nn.Module):
                 mean, std = statistic[i].strip().split(' ')
                 print(new_emb[i].size())
                 print(float(mean), float(std))
-                new_emb[i] = torch.normal(float(mean), float(std), size=(50)).squeeze()
+                new_emb[i] = torch.normal(float(mean), float(std), size=(1,50)).squeeze()
             new_emb = new_emb.transpose(1,0)
             self.prompt = torch.nn.Parameter(new_emb) 
             print(self.prompt)