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

auto frontend

Browse Source
This commit is contained in:
游雁 2024-06-03 17:13:48 +08:00
parent 99ecaca869
commit 1fcb757216
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/auto/auto_frontend.py
View File

@ -52,7 +52,7 @@ class AutoFrontend:
key_list, data_list = prepare_data_iterator(input, input_len=input_len)
batch_size = kwargs.get("batch_size", 1)
device = kwargs.get("device", "cpu")
device = kwargs.get("device", "cuda")
if device == "cpu":
batch_size = 1