From 2868fe3df4e92a6ae3e327faf6e57ea492e04124 Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期四, 16 三月 2023 19:24:21 +0800
Subject: [PATCH] Merge branch 'main' into dev_dzh

---
 funasr/bin/lm_inference.py |    5 +++--
 1 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/funasr/bin/lm_inference.py b/funasr/bin/lm_inference.py
index 909cb02..15c56ca 100644
--- a/funasr/bin/lm_inference.py
+++ b/funasr/bin/lm_inference.py
@@ -129,8 +129,9 @@
         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
 

--
Gitblit v1.9.1