Fangjun Kuang
Committed by GitHub

Fix #510 (#513)

@@ -862,6 +862,7 @@ def create_recognizer(args) -> sherpa_onnx.OfflineRecognizer: @@ -862,6 +862,7 @@ def create_recognizer(args) -> sherpa_onnx.OfflineRecognizer:
862 max_active_paths=args.max_active_paths, 862 max_active_paths=args.max_active_paths,
863 hotwords_file=args.hotwords_file, 863 hotwords_file=args.hotwords_file,
864 hotwords_score=args.hotwords_score, 864 hotwords_score=args.hotwords_score,
  865 + provider=args.provider,
865 ) 866 )
866 elif args.paraformer: 867 elif args.paraformer:
867 assert len(args.nemo_ctc) == 0, args.nemo_ctc 868 assert len(args.nemo_ctc) == 0, args.nemo_ctc
@@ -879,6 +880,7 @@ def create_recognizer(args) -> sherpa_onnx.OfflineRecognizer: @@ -879,6 +880,7 @@ def create_recognizer(args) -> sherpa_onnx.OfflineRecognizer:
879 sample_rate=args.sample_rate, 880 sample_rate=args.sample_rate,
880 feature_dim=args.feat_dim, 881 feature_dim=args.feat_dim,
881 decoding_method=args.decoding_method, 882 decoding_method=args.decoding_method,
  883 + provider=args.provider,
882 ) 884 )
883 elif args.nemo_ctc: 885 elif args.nemo_ctc:
884 assert len(args.wenet_ctc) == 0, args.wenet_ctc 886 assert len(args.wenet_ctc) == 0, args.wenet_ctc
@@ -895,6 +897,7 @@ def create_recognizer(args) -> sherpa_onnx.OfflineRecognizer: @@ -895,6 +897,7 @@ def create_recognizer(args) -> sherpa_onnx.OfflineRecognizer:
895 sample_rate=args.sample_rate, 897 sample_rate=args.sample_rate,
896 feature_dim=args.feat_dim, 898 feature_dim=args.feat_dim,
897 decoding_method=args.decoding_method, 899 decoding_method=args.decoding_method,
  900 + provider=args.provider,
898 ) 901 )
899 elif args.wenet_ctc: 902 elif args.wenet_ctc:
900 assert len(args.whisper_encoder) == 0, args.whisper_encoder 903 assert len(args.whisper_encoder) == 0, args.whisper_encoder
@@ -910,6 +913,7 @@ def create_recognizer(args) -> sherpa_onnx.OfflineRecognizer: @@ -910,6 +913,7 @@ def create_recognizer(args) -> sherpa_onnx.OfflineRecognizer:
910 sample_rate=args.sample_rate, 913 sample_rate=args.sample_rate,
911 feature_dim=args.feat_dim, 914 feature_dim=args.feat_dim,
912 decoding_method=args.decoding_method, 915 decoding_method=args.decoding_method,
  916 + provider=args.provider,
913 ) 917 )
914 elif args.whisper_encoder: 918 elif args.whisper_encoder:
915 assert len(args.tdnn_model) == 0, args.tdnn_model 919 assert len(args.tdnn_model) == 0, args.tdnn_model
@@ -925,6 +929,7 @@ def create_recognizer(args) -> sherpa_onnx.OfflineRecognizer: @@ -925,6 +929,7 @@ def create_recognizer(args) -> sherpa_onnx.OfflineRecognizer:
925 language=args.whisper_language, 929 language=args.whisper_language,
926 task=args.whisper_task, 930 task=args.whisper_task,
927 tail_paddings=args.whisper_tail_paddings, 931 tail_paddings=args.whisper_tail_paddings,
  932 + provider=args.provider,
928 ) 933 )
929 elif args.tdnn_model: 934 elif args.tdnn_model:
930 assert_file_exists(args.tdnn_model) 935 assert_file_exists(args.tdnn_model)
@@ -936,6 +941,7 @@ def create_recognizer(args) -> sherpa_onnx.OfflineRecognizer: @@ -936,6 +941,7 @@ def create_recognizer(args) -> sherpa_onnx.OfflineRecognizer:
936 feature_dim=args.feat_dim, 941 feature_dim=args.feat_dim,
937 num_threads=args.num_threads, 942 num_threads=args.num_threads,
938 decoding_method=args.decoding_method, 943 decoding_method=args.decoding_method,
  944 + provider=args.provider,
939 ) 945 )
940 else: 946 else:
941 raise ValueError("Please specify at least one model") 947 raise ValueError("Please specify at least one model")