(output_ids: torch.Tensor)
| 269 | |
| 270 | |
| 271 | def decode(output_ids: torch.Tensor): |
| 272 | output_ids = output_ids.cpu().numpy() |
| 273 | return _tokenizer.decode(output_ids) |
| 274 | |
| 275 | |
| 276 | def tokenize(texts: Union[str, List[str]], context_length: int = DEFAULT_CONTEXT_LENGTH) -> torch.LongTensor: |