(self, tokenizer, return_source_length=False)
| 126 | |
| 127 | class WikiLoader(ConditionalLoader): |
| 128 | def __init__(self, tokenizer, return_source_length=False): |
| 129 | super(WikiLoader, self).__init__(tokenizer, return_source_length) |
| 130 | self.task_name = 'wiki_alignment' |
| 131 | |
| 132 | @staticmethod |
| 133 | def add_prompt(example): |