diff --git a/src/maxtext/layers/decoders.py b/src/maxtext/layers/decoders.py index d2deb7192a..708b826098 100644 --- a/src/maxtext/layers/decoders.py +++ b/src/maxtext/layers/decoders.py @@ -1026,9 +1026,6 @@ def __call__( current_broadcast_args.extend([None, None, None, attention_metadata]) current_in_axes_tuple.extend([nn.broadcast] * 4) - max_logging.info(f"DEBUG: len(current_broadcast_args)={len(current_broadcast_args)}") - max_logging.info(f"DEBUG: current_broadcast_args={[type(a) for a in current_broadcast_args]}") - final_carry, _ = self.scan_decoder_layers( cfg, RemattedBlockLayer,