소스 검색

Remove unnecessary apply_padding_mask operations.

Kaushik Ram Sadagopan 2 년 전
부모
커밋
a55c74eb41
2개의 변경된 파일0개의 추가작업 그리고 6개의 파일을 삭제
  1. 0 4
      src/seamless_communication/models/unity/length_regulator.py
  2. 0 2
      src/seamless_communication/models/unity/nar_decoder_layer.py

+ 0 - 4
src/seamless_communication/models/unity/length_regulator.py

@@ -133,13 +133,9 @@ class VariancePredictor(Module):
 
         seqs = self.dropout_module(seqs)
 
-        seqs = apply_padding_mask(seqs, padding_mask)
-
         # (N, S, H) -> (N, S, 1) -> (N, S)
         seqs = self.proj(seqs).squeeze(dim=2)
 
-        seqs = apply_padding_mask(seqs, padding_mask)
-
         return seqs
 
 

+ 0 - 2
src/seamless_communication/models/unity/nar_decoder_layer.py

@@ -103,8 +103,6 @@ class Conv1dBlock(Module):
         # (N, M, S) -> (N, S, M)
         seqs = seqs.transpose(1, 2)
 
-        seqs = apply_padding_mask(seqs, padding_mask)
-
         return seqs