Support specifying provider for python examples (#244)

This commit is contained in:
Fangjun Kuang
2023-08-09 10:00:34 +08:00
committed by GitHub
parent 6235cb9f1b
commit aeb112dd06
4 changed files with 35 additions and 1 deletions

View File

@@ -129,6 +129,13 @@ def add_model_args(parser: argparse.ArgumentParser):
help="Feature dimension of the model",
)
parser.add_argument(
"--provider",
type=str,
default="cpu",
help="Valid values: cpu, cuda, coreml",
)
def add_decoding_args(parser: argparse.ArgumentParser):
parser.add_argument(
@@ -301,6 +308,7 @@ def create_recognizer(args) -> sherpa_onnx.OnlineRecognizer:
rule1_min_trailing_silence=args.rule1_min_trailing_silence,
rule2_min_trailing_silence=args.rule2_min_trailing_silence,
rule3_min_utterance_length=args.rule3_min_utterance_length,
provider=args.provider,
)
return recognizer