speech_asr 3 년 전
부모
커밋
429ea5d378
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      funasr/bin/eend_ola_inference.py

+ 1 - 1
funasr/bin/eend_ola_inference.py

@@ -145,7 +145,7 @@ def inference_modelscope(
         output_dir: Optional[str] = None,
         batch_size: int = 1,
         dtype: str = "float32",
-        ngpu: int = 0,
+        ngpu: int = 1,
         num_workers: int = 0,
         log_level: Union[int, str] = "INFO",
         key_file: Optional[str] = None,