北念 2 лет назад
Родитель
Сommit
72a0600129
1 измененных файлов с 6 добавлено и 2 удалено
  1. 6 2
      funasr/bin/asr_inference_launch.py

+ 6 - 2
funasr/bin/asr_inference_launch.py

@@ -2056,8 +2056,12 @@ def inference_whisper(
 
     ncpu = kwargs.get("ncpu", 1)
     torch.set_num_threads(ncpu)
-    language = param_dict.get("language", None)
-    task = param_dict.get("task", "transcribe")
+    if param_dict:
+        language = param_dict.get("language", None)
+        task = param_dict.get("task", "transcribe")
+    else:
+        language = None
+        task = "transcribe"
     if batch_size > 1:
         raise NotImplementedError("batch decoding is not implemented")
     if word_lm_train_config is not None: