def uncollate(self, generated_tokens: Any) -> Any:
    pred_str = self.tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
    pred_str = [str.strip(s) for s in pred_str]
    return pred_str
def uncollate(self, generated_tokens: Any) -> Any:
    pred_str = self.tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
    pred_str = [str.strip(s) for s in pred_str]
    return pred_str