speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add ContextualParaformer

Browse Source
This commit is contained in:
北念 2023-02-09 20:08:44 +08:00
parent 5231b54af8
commit 212b20d47f
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
funasr/bin/asr_inference_paraformer.py
View File

@ -428,6 +428,7 @@ def inference_modelscope(
format="%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s",
)
hotword_list_or_file = param_dict['hotword']
if ngpu >= 1 and torch.cuda.is_available():
device = "cuda"
else: