def METHOD_NAME(self, input_ids):
    if self.mode == 'codegeex-13b':
        text = self.tokenizer.decode(input_ids, skip_special_tokens=False)
        output_code = decode_whitespaces(text, self.start_extra_id,
                                         self.max_len)
    elif self.mode == 'codegeex-python-13b':
        input_ids = [self.code_dict.decode_tokens(input_ids.tolist()[0])]
        text = self.tokenizer.decode(input_ids, skip_special_tokens=False)
        output_code = decode_whitespaces(text, self.start_extra_id,
                                         self.max_len)
    return output_code