This commit is contained in:
游雁 2024-03-19 11:56:57 +08:00
parent 5a637c6995
commit 60a7d39f39

View File

@ -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)