funasr1.0

游雁

2024-01-13 bdfd27b9e96bd55c449953bb577e1d4deeaf11c9

 funasr/tokenizer/char_tokenizer.py

@@ -4,17 +4,19 @@
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()