Update custom_heads_flash_t5.py

custom_heads_flash_t5.py

@@ -257,6 +257,7 @@ class FlashT5ForQuestionAnswering(FlashT5PreTrainedModel):
             Labels for position (index) of the end of the labelled span for computing the token classification loss.
             Positions are clamped to the length of the sequence (*sequence_length*). Position outside of the sequence
             are not taken into account for computing the loss.
+
         Returns:
         """
         return_dict = return_dict if return_dict is not None else self.config.use_return_dict
@@ -344,9 +345,12 @@ class FlashT5ForQuestionAnswering(FlashT5PreTrainedModel):
     ) -> Union[Tuple, QuestionAnsweringModelOutput]:
         r"""
         Returns:
+
         Example:
+
         ```python
         >>> from transformers import AutoTokenizer, MTxEncoderForQuestionAnswering
+
         >>> tokenizer = AutoTokenizer.from_pretrained("MTx-small")
         >>> model = MTxEncoderForQuestionAnswering.from_pretrained("MTx-small")
         >>> input_ids = tokenizer(