fix lint

wenet-e2e · Nov 13, 2023 · f8fea10 · f8fea10
1 parent 5b6d758
commit f8fea10
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/wenet/paraformer/paraformer.py b/wenet/paraformer/paraformer.py
@@ -87,8 +87,8 @@ def __init__(self,
             del self.decoder.embed
         # NOTE(Mddct): add eos in tail of labels for predictor
         # eg:
-        #.    gt:         你 好 we@@ net
-        #.    labels:     你 好 we@@ net eos
+        #    gt:         你 好 we@@ net
+        #    labels:     你 好 we@@ net eos
         self.add_eos = add_eos
 
     @torch.jit.ignore(drop=True)
@@ -165,7 +165,6 @@ def _sampler(self, encoder_out, encoder_out_mask, ys_pad, ys_pad_lens,
         device = encoder_out.device
         B, _ = ys_pad.size()
 
-        # update from: https://github.com/alibaba-damo-academy/FunASR/blob/main/funasr/models/e2e_asr_paraformer.py#L526
         tgt_mask = make_non_pad_mask(ys_pad_lens)
         ys_pad_embed = self.embed(ys_pad)  # [B, T, L]
         with torch.no_grad():
@@ -193,8 +192,8 @@ def _sampler(self, encoder_out, encoder_out_mask, ys_pad, ys_pad_lens,
             input_mask = input_mask * tgt_mask
             input_mask_expand = input_mask.unsqueeze(2)  # [B, T, 1]
 
-        sematic_embeds = torch.where(input_mask_expand == 1, ys_pad_embed,
-                                     pre_acoustic_embeds)
+        sematic_embeds = torch.where(input_mask_expand == 1,
+                                     pre_acoustic_embeds, ys_pad_embed)
         # zero out the paddings
         return sematic_embeds * tgt_mask.unsqueeze(2)