| | |
| | | |
| | | ncpu = kwargs.get("ncpu", 1) |
| | | torch.set_num_threads(ncpu) |
| | | if param_dict: |
| | | language = param_dict.get("language", None) |
| | | task = param_dict.get("task", "transcribe") |
| | | else: |
| | | language = None |
| | | task = "transcribe" |
| | | if batch_size > 1: |
| | | raise NotImplementedError("batch decoding is not implemented") |
| | | if word_lm_train_config is not None: |
| | |
| | | penalty=penalty, |
| | | nbest=nbest, |
| | | streaming=streaming, |
| | | language=language, |
| | | task=task, |
| | | ) |
| | | logging.info("speech2text_kwargs: {}".format(speech2text_kwargs)) |
| | | speech2text = Speech2TextWhisper(**speech2text_kwargs) |