From 9595a9432fadfbdacd4e6897f6b9a83957699558 Mon Sep 17 00:00:00 2001
From: seanzhang-zhichen <74812416+seanzhang-zhichen@users.noreply.github.com>
Date: 星期二, 05 三月 2024 17:42:14 +0800
Subject: [PATCH] modify paraformer train doc (#1427)
---
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 ec3c3f3..70d09df 100644
--- a/funasr/auto/auto_model.py
+++ b/funasr/auto/auto_model.py
@@ -143,7 +143,7 @@
def build_model(self, **kwargs):
assert "model" in kwargs
if "model_conf" not in kwargs:
- logging.info("download models from model hub: {}".format(kwargs.get("model_hub", "ms")))
+ logging.info("download models from model hub: {}".format(kwargs.get("hub", "ms")))
kwargs = download_model(**kwargs)
set_all_random_seed(kwargs.get("seed", 0))
@@ -180,7 +180,7 @@
# build model
model_class = tables.model_classes.get(kwargs["model"])
- model = model_class(**kwargs, **kwargs["model_conf"], vocab_size=vocab_size)
+ model = model_class(**kwargs, **kwargs.get("model_conf", {}), vocab_size=vocab_size)
model.to(device)
# init_param
--
Gitblit v1.9.1