diff --git a/paddlespeech/cli/README.md b/paddlespeech/cli/README.md index bd6572f1..56afb939 100644 --- a/paddlespeech/cli/README.md +++ b/paddlespeech/cli/README.md @@ -6,4 +6,4 @@ `paddlespeech help` ## ASR - `paddlespeech asr --input ./test_audio.wav --device gpu` + `paddlespeech asr --input ./test_audio.wav` diff --git a/paddlespeech/cli/asr/infer.py b/paddlespeech/cli/asr/infer.py index 60516380..e5c64e9a 100644 --- a/paddlespeech/cli/asr/infer.py +++ b/paddlespeech/cli/asr/infer.py @@ -87,7 +87,7 @@ class ASRExecutor(BaseExecutor): self.parser.add_argument( '--device', type=str, - default='cpu', + default=paddle.get_device(), help='Choose device to execute model inference.') def _get_pretrained_path(self, tag: str) -> os.PathLike: