From 00ea1186f96e6732e2edb4fab6c0ed6896e3b352 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 19 十二月 2023 22:53:18 +0800
Subject: [PATCH] funasr2
---
funasr/bin/inference.py | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/funasr/bin/inference.py b/funasr/bin/inference.py
index 09e28f3..fd884cd 100644
--- a/funasr/bin/inference.py
+++ b/funasr/bin/inference.py
@@ -81,7 +81,7 @@
class AutoModel:
def __init__(self, **kwargs):
- registry_tables.print_register_tables()
+ registry_tables.print()
assert "model" in kwargs
if "model_conf" not in kwargs:
logging.info("download models from model hub: {}".format(kwargs.get("model_hub", "ms")))
@@ -108,6 +108,7 @@
frontend_class = registry_tables.frontend_classes.get(frontend.lower())
frontend = frontend_class(**kwargs["frontend_conf"])
kwargs["frontend"] = frontend
+ kwargs["input_size"] = frontend.output_size()
# build model
model_class = registry_tables.model_classes.get(kwargs["model"].lower())
--
Gitblit v1.9.1