bugfix

Author: wtmlon

paddlenlp/data/data_collator.py

@@ -402,7 +402,7 @@ def __call__(self, features, return_tensors=None):
                 else:
                     feature["labels"] = np.concatenate([remainder, feature["labels"]]).astype(np.int64)
         if use_attn_mask_startend_row_indices is not None:
-            if self.max_label_length is not None:
+            if self.max_length is not None:
                 max_length = self.max_length
             else:
                 max_length = max(len(l) for l in use_attn_mask_startend_row_indices)