From 08e6f946aa4c02e3ba3cebb47af2ac3cd5abe97b Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 05 二月 2024 11:09:59 +0800
Subject: [PATCH] disable_log
---
funasr/auto/auto_model.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/auto/auto_model.py b/funasr/auto/auto_model.py
index 3986a11..ae47d35 100644
--- a/funasr/auto/auto_model.py
+++ b/funasr/auto/auto_model.py
@@ -88,7 +88,7 @@
class AutoModel:
def __init__(self, **kwargs):
- if kwargs.get("disable_log", False):
+ if not kwargs.get("disable_log", False):
tables.print()
model, kwargs = self.build_model(**kwargs)
@@ -228,7 +228,7 @@
data_batch = data_list[beg_idx:end_idx]
key_batch = key_list[beg_idx:end_idx]
batch = {"data_in": data_batch, "key": key_batch}
- if (end_idx - beg_idx) == 1 and isinstance(data_batch[0], torch.Tensor): # fbank
+ if (end_idx - beg_idx) == 1 and kwargs.get("data_type", None) == "fbank": # fbank
batch["data_in"] = data_batch[0]
batch["data_lengths"] = input_len
--
Gitblit v1.9.1