(self, model_name: str)
| 7 | |
| 8 | class Tokenizer: |
| 9 | def __init__(self, model_name: str): |
| 10 | self._tokenizer = transformers.AutoTokenizer.from_pretrained( |
| 11 | model_name, |
| 12 | legacy=False, |
| 13 | model_max_length=512, |
| 14 | ) |
| 15 | self._decoder_start_id = 0 |
| 16 | |
| 17 | @property |
| 18 | def eos_id(self) -> int: |
nothing calls this directly
no test coverage detected