正在显示
1 个修改的文件
包含
6 行增加
和
0 行删除
| @@ -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") |
-
请 注册 或 登录 后发表评论