| | |
| | | from typing import Union |
| | | import warnings |
| | | |
| | | from funasr.tokenizer.abs_tokenizer import BaseTokenizer |
| | | from funasr.register import tables |
| | | |
| | | from funasr.tokenizer.abs_tokenizer import AbsTokenizer |
| | | |
| | | |
| | | class CharTokenizer(AbsTokenizer): |
| | | @tables.register("tokenizer_classes", "CharTokenizer") |
| | | class CharTokenizer(BaseTokenizer): |
| | | def __init__( |
| | | self, |
| | | non_linguistic_symbols: Union[Path, str, Iterable[str]] = None, |
| | | space_symbol: str = "<space>", |
| | | remove_non_linguistic_symbols: bool = False, |
| | | **kwargs, |
| | | ): |
| | | super().__init__(**kwargs) |
| | | self.space_symbol = space_symbol |
| | | if non_linguistic_symbols is None: |
| | | self.non_linguistic_symbols = set() |
| | |
| | | |
| | | def tokens2text(self, tokens: Iterable[str]) -> str: |
| | | tokens = [t if t != self.space_symbol else " " for t in tokens] |
| | | return "".join(tokens) |
| | | return "".join(tokens) |