| | |
|---|
| | | from typing import Union |
|---|
| | | |
|---|
| | | import torch |
|---|
| | | from typeguard import check_argument_types |
|---|
| | | |
|---|
| | | from funasr.bin.punc_infer import Text2Punc, Text2PuncVADRealtime |
|---|
| | | from funasr.torch_utils.set_all_random_seed import set_all_random_seed |
|---|
| | |
|---|
| | | param_dict: dict = None, |
|---|
| | | **kwargs, |
|---|
| | | ): |
|---|
| | | assert check_argument_types() |
|---|
| | | logging.basicConfig( |
|---|
| | | level=log_level, |
|---|
| | | format="%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s", |
|---|
| | |
|---|
| | | param_dict: dict = None, |
|---|
| | | **kwargs, |
|---|
| | | ): |
|---|
| | | assert check_argument_types() |
|---|
| | | ncpu = kwargs.get("ncpu", 1) |
|---|
| | | torch.set_num_threads(ncpu) |
|---|
| | | |
|---|
