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.