| | |
|---|
| | | if not model_path.is_file(): |
|---|
| | | raise FileExistsError(f'{model_path} is not a file.') |
|---|
| | | |
|---|
| | | def split_to_mini_sentence(words: list, word_limit: int = 20): |
|---|
| | | assert word_limit > 1 |
|---|
| | | if len(words) <= word_limit: |
|---|
| | | return [words] |
|---|
| | | sentences = [] |
|---|
| | | length = len(words) |
|---|
| | | sentence_len = length // word_limit |
|---|
| | | for i in range(sentence_len): |
|---|
| | | sentences.append(words[i * word_limit:(i + 1) * word_limit]) |
|---|
| | | if length % word_limit > 0: |
|---|
| | | sentences.append(words[sentence_len * word_limit:]) |
|---|
| | | return sentences |
|---|
| | | |
|---|
| | | |
|---|
| | | def read_yaml(yaml_path: Union[str, Path]) -> Dict: |
|---|
| | | if not Path(yaml_path).exists(): |
|---|
| | |
|---|
| | | |
|---|
| | | |
|---|
| | | @functools.lru_cache() |
|---|
| | | def get_logger(name='rapdi_paraformer'): |
|---|
| | | def get_logger(name='funasr_onnx'): |
|---|
| | | """Initialize and get a logger by name. |
|---|
| | | If the logger has not been initialized, this method will initialize the |
|---|
| | | logger by adding one or two handlers, otherwise the initialized logger will |
|---|
