From 60a7d39f3939e9e901049a1f6f11b4660c4eab25 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Tue, 19 Mar 2024 11:56:57 +0800 Subject: [PATCH] vad conf --- funasr/auto/auto_model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/funasr/auto/auto_model.py b/funasr/auto/auto_model.py index d0444557d..bba44e792 100644 --- a/funasr/auto/auto_model.py +++ b/funasr/auto/auto_model.py @@ -187,7 +187,8 @@ class AutoModel: kwargs["frontend"] = frontend # build model model_class = tables.model_classes.get(kwargs["model"]) - model_conf = kwargs.get("model_conf", {}) + model_conf = {} + deep_update(model_conf, kwargs.get("model_conf", {})) deep_update(model_conf, kwargs) model = model_class(**model_conf, vocab_size=vocab_size) model.to(device)