Adding attention mask into MLM

To add to this, my code is killed following the message above, and no config.json file is created in the sentence transformer