In the dataset to_tf_dataset, token_type_ids is passed — RoBERTa doesn't need it. Propose a warning to check model type before blindly passing.