From 0d155388355f842eb4b9c6d3782798a206819578 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 17 二月 2023 14:51:47 +0800
Subject: [PATCH] Merge pull request #133 from alibaba-damo-academy/dev_lyb

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