Skip to content

Commit

Permalink
keyword device
Browse files Browse the repository at this point in the history
  • Loading branch information
manickavela29 committed Jul 4, 2024
1 parent 6ab0567 commit 04a9d8d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions sherpa-onnx/python/sherpa_onnx/keyword_spotter.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ def __init__(
keywords_threshold: float = 0.25,
num_trailing_blanks: int = 1,
provider: str = "cpu",
device: int = 1,
device: int = 0,
):
"""
Please refer to
Expand Down Expand Up @@ -89,7 +89,6 @@ def __init__(
onnxruntime execution providers. Valid values are: cpu, cuda, coreml.
device:
onnxruntime cuda device index.
"""
_assert_file_exists(tokens)
_assert_file_exists(encoder)
Expand All @@ -108,6 +107,7 @@ def __init__(
provider=provider,
device = device,
)

model_config = OnlineModelConfig(
transducer=transducer_config,
tokens=tokens,
Expand Down

0 comments on commit 04a9d8d

Please sign in to comment.