HoneyTian commited on
Commit
b6416e6
·
1 Parent(s): 8692b5e
Files changed (1) hide show
  1. toolbox/k2_sherpa/nn_models.py +4 -0
toolbox/k2_sherpa/nn_models.py CHANGED
@@ -379,6 +379,10 @@ def load_recognizer(local_model_dir: Path,
379
  decoding_method=decoding_method,
380
  **kwargs_
381
  )
 
 
 
 
382
  else:
383
  raise NotImplementedError("loader not support: {}".format(loader))
384
  return recognizer
 
379
  decoding_method=decoding_method,
380
  **kwargs_
381
  )
382
+ elif loader == "load_sherpa_offline_recognizer_from_whisper":
383
+ recognizer = load_sherpa_offline_recognizer_from_whisper(
384
+ **kwargs_
385
+ )
386
  else:
387
  raise NotImplementedError("loader not support: {}".format(loader))
388
  return recognizer