shixian.shi преди 2 години
родител
ревизия
f2eb55ded7
променени са 1 файла, в които са добавени 1 реда и са изтрити 0 реда
  1. 1 0
      funasr/models/e2e_asr_contextual_paraformer.py

+ 1 - 0
funasr/models/e2e_asr_contextual_paraformer.py

@@ -350,6 +350,7 @@ class NeatContextualParaformer(Paraformer):
             else:
                 hw_embed = self.bias_embed(hw_list_pad)
             hw_embed, (h_n, _) = self.bias_encoder(hw_embed)
+            hw_embed = h_n.repeat(encoder_out.shape[0], 1, 1)
         else:
             hw_lengths = [len(i) for i in hw_list]
             hw_list_pad = pad_list([torch.Tensor(i).long() for i in hw_list], 0).to(encoder_out.device)