feat: support adding attention mask when not present for pretokenized data

[kmehant]

When in the case of pretokenized datasets, we should provide the feature on computing attention mask on the go when not part of the provided datasets. This can opinionated and up for discussion.