Attention layers now allow None paddings, remove assert.

PiperOrigin-RevId: 685751950
tensorflow · Oct 14, 2024 · 48e5034 · 48e5034
1 parent b821d00
commit 48e5034
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/lingvo/core/layers_with_attention.py b/lingvo/core/layers_with_attention.py
@@ -1500,7 +1500,6 @@ def FProp(
 
     if p.has_aux_atten:
       assert aux_vecs is not None
-      assert aux_paddings is not None
       with tf.name_scope('aux_atten'):
         atten_vec, atten_prob = self.atten.FProp(theta.atten, atten_vec,
                                                  aux_paddings, aux_vecs,
@@ -1550,7 +1549,6 @@ def ExtendStep(self,
 
     if p.has_aux_atten:
       assert aux_vecs is not None
-      assert aux_paddings is not None
 
     batch_size = py_utils.GetShape(source_vecs)[0]