From 364dc9740954c93499abaee78f38e6e8727adea2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Sat, 23 May 2026 21:54:27 +0800 Subject: [PATCH] fix: propagate hub parameter to vad/punc/spk sub-model builds (#2859) When user passes hub='hf', the parameter was lost when building sub-models (vad_model, punc_model, spk_model), causing them to default to ModelScope and fail with 404 for HuggingFace model IDs. Now hub is forwarded via setdefault so sub-models inherit the parent's hub preference unless explicitly overridden. --- funasr/auto/auto_model.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/funasr/auto/auto_model.py b/funasr/auto/auto_model.py index c343256a8..46d8b42b7 100644 --- a/funasr/auto/auto_model.py +++ b/funasr/auto/auto_model.py @@ -228,6 +228,8 @@ class AutoModel: vad_kwargs["model_revision"] = kwargs.get("vad_model_revision", "master") vad_kwargs["device"] = kwargs["device"] vad_kwargs.setdefault("ncpu", kwargs.get("ncpu", 4)) + if "hub" in kwargs: + vad_kwargs.setdefault("hub", kwargs["hub"]) vad_model, vad_kwargs = self.build_model(**vad_kwargs) # if punc_model is not None, build punc model else None @@ -239,6 +241,8 @@ class AutoModel: punc_kwargs["model_revision"] = kwargs.get("punc_model_revision", "master") punc_kwargs["device"] = kwargs["device"] punc_kwargs.setdefault("ncpu", kwargs.get("ncpu", 4)) + if "hub" in kwargs: + punc_kwargs.setdefault("hub", kwargs["hub"]) punc_model, punc_kwargs = self.build_model(**punc_kwargs) # if spk_model is not None, build spk model else None @@ -253,6 +257,8 @@ class AutoModel: spk_kwargs["model_revision"] = kwargs.get("spk_model_revision", "master") spk_kwargs["device"] = kwargs["device"] spk_kwargs.setdefault("ncpu", kwargs.get("ncpu", 4)) + if "hub" in kwargs: + spk_kwargs.setdefault("hub", kwargs["hub"]) spk_model, spk_kwargs = self.build_model(**spk_kwargs) self.cb_model = ClusterBackend(**cb_kwargs).to(kwargs["device"]) spk_mode = kwargs.get("spk_mode", "punc_segment")