瀏覽代碼

Merge pull request #133 from alibaba-damo-academy/dev_lyb

fix bug: output_dir
zhifu gao 3 年之前
父節點
當前提交
0d15538835
共有 1 個文件被更改,包括 3 次插入2 次删除
  1. 3 2
      funasr/bin/lm_inference.py

+ 3 - 2
funasr/bin/lm_inference.py

@@ -129,8 +129,9 @@ def inference_modelscope(
         param_dict: dict = None,
     ):
         results = []
-        if output_dir_v2 is not None:
-            writer = DatadirWriter(output_dir_v2)
+        output_path = output_dir_v2 if output_dir_v2 is not None else output_dir
+        if output_path is not None:
+            writer = DatadirWriter(output_path)
         else:
             writer = None