Эх сурвалжийг харах

Remove unnecessary apply_padding_mask operations.

Kaushik Ram Sadagopan 2 жил өмнө
parent
commit
a55c74eb41

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

@@ -133,13 +133,9 @@ class VariancePredictor(Module):
 
 
         seqs = self.dropout_module(seqs)
         seqs = self.dropout_module(seqs)
 
 
-        seqs = apply_padding_mask(seqs, padding_mask)
-
         # (N, S, H) -> (N, S, 1) -> (N, S)
         # (N, S, H) -> (N, S, 1) -> (N, S)
         seqs = self.proj(seqs).squeeze(dim=2)
         seqs = self.proj(seqs).squeeze(dim=2)
 
 
-        seqs = apply_padding_mask(seqs, padding_mask)
-
         return seqs
         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)
         # (N, M, S) -> (N, S, M)
         seqs = seqs.transpose(1, 2)
         seqs = seqs.transpose(1, 2)
 
 
-        seqs = apply_padding_mask(seqs, padding_mask)
-
         return seqs
         return seqs