Why is intra-document masking only applied for long-context training and not midtraining?

I noticed that only the long-context stages seem to apply intra-document masking with `generate_doc_lengths=True` in the NumpyDataset config. Is there any reason that midtraining does not also benefit from reducing cross-document attention signals? Thank you!