Skip to content

Commit 364dc97

Browse files
committed
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.
1 parent c8ebbc7 commit 364dc97

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

funasr/auto/auto_model.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -228,6 +228,8 @@ def __init__(self, **kwargs):
228228
vad_kwargs["model_revision"] = kwargs.get("vad_model_revision", "master")
229229
vad_kwargs["device"] = kwargs["device"]
230230
vad_kwargs.setdefault("ncpu", kwargs.get("ncpu", 4))
231+
if "hub" in kwargs:
232+
vad_kwargs.setdefault("hub", kwargs["hub"])
231233
vad_model, vad_kwargs = self.build_model(**vad_kwargs)
232234

233235
# if punc_model is not None, build punc model else None
@@ -239,6 +241,8 @@ def __init__(self, **kwargs):
239241
punc_kwargs["model_revision"] = kwargs.get("punc_model_revision", "master")
240242
punc_kwargs["device"] = kwargs["device"]
241243
punc_kwargs.setdefault("ncpu", kwargs.get("ncpu", 4))
244+
if "hub" in kwargs:
245+
punc_kwargs.setdefault("hub", kwargs["hub"])
242246
punc_model, punc_kwargs = self.build_model(**punc_kwargs)
243247

244248
# if spk_model is not None, build spk model else None
@@ -253,6 +257,8 @@ def __init__(self, **kwargs):
253257
spk_kwargs["model_revision"] = kwargs.get("spk_model_revision", "master")
254258
spk_kwargs["device"] = kwargs["device"]
255259
spk_kwargs.setdefault("ncpu", kwargs.get("ncpu", 4))
260+
if "hub" in kwargs:
261+
spk_kwargs.setdefault("hub", kwargs["hub"])
256262
spk_model, spk_kwargs = self.build_model(**spk_kwargs)
257263
self.cb_model = ClusterBackend(**cb_kwargs).to(kwargs["device"])
258264
spk_mode = kwargs.get("spk_mode", "punc_segment")

0 commit comments

Comments
 (0)