update

toolbox/k2_sherpa/nn_models.py

@@ -23,6 +23,7 @@ model_map = {
             "tokens_file": "units.txt",
             "sub_folder": ".",
             "loader": "load_sherpa_offline_recognizer",
+            "normalize_samples": False,
         },
         {
             "repo_id": "csukuangfj/sherpa-onnx-paraformer-zh-2024-03-09",
@@ -44,6 +45,7 @@ model_map = {
             "tokens_file": "tokens.txt",
             "sub_folder": ".",
             "loader": "load_sherpa_offline_recognizer",
+            "normalize_samples": True,
         }
     ],
     "English": [
@@ -106,8 +108,9 @@ def load_sherpa_offline_recognizer(nn_model_file: str,
                                    decoding_method: str = "greedy_search",
                                    num_mel_bins: int = 80,
                                    frame_dither: int = 0,
+                                   normalize_samples: bool = False,
                                    ):
-    feat_config = sherpa.FeatureConfig(normalize_samples=False)
+    feat_config = sherpa.FeatureConfig(normalize_samples=normalize_samples)
     feat_config.fbank_opts.frame_opts.samp_freq = sample_rate
     feat_config.fbank_opts.mel_opts.num_bins = num_mel_bins
     feat_config.fbank_opts.frame_opts.dither = frame_dither