From 28a19dbc4e85d3b8a4ec2ef7483bba64d422b43f Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期三, 12 四月 2023 18:03:06 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/main' into dev_aky

---
 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