(self, input_ids)
| 56 | return input_ids |
| 57 | |
| 58 | def decode_code(self, input_ids): |
| 59 | if self.mode == 'codegeex-13b': |
| 60 | text = self.tokenizer.decode(input_ids, skip_special_tokens=False, verbose=False) |
| 61 | output_code = decode_whitespaces(text, self.start_extra_id, self.max_len) |
| 62 | |
| 63 | return output_code |