shixian.shi 2 лет назад
Родитель
Сommit
b7befd113a
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      funasr/bin/asr_inference_launch.py

+ 2 - 1
funasr/bin/asr_inference_launch.py

@@ -255,9 +255,10 @@ def inference_paraformer(
     if param_dict is not None:
     if param_dict is not None:
         hotword_list_or_file = param_dict.get('hotword')
         hotword_list_or_file = param_dict.get('hotword')
         export_mode = param_dict.get("export_mode", False)
         export_mode = param_dict.get("export_mode", False)
+        clas_scale = param_dict.get('clas_scale', 1.0)
     else:
     else:
         hotword_list_or_file = None
         hotword_list_or_file = None
-    clas_scale = param_dict.get('clas_scale', 1.0)
+        clas_scale = 1.0
 
 
     if kwargs.get("device", None) == "cpu":
     if kwargs.get("device", None) == "cpu":
         ngpu = 0
         ngpu = 0