From 40eefbe376bd2e846bd9c451636f4e42ba1bf8bf Mon Sep 17 00:00:00 2001
From: zhuzizyf <42790740+zhuzizyf@users.noreply.github.com>
Date: 星期三, 12 四月 2023 17:18:19 +0800
Subject: [PATCH] Merge pull request #1 from zhuzizyf/fix-dataset-bug
---
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