funasr/auto/auto_model.py
@@ -187,7 +187,8 @@ 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)
