add check for sr other than 8k or 16k

microsoft · Feb 27, 2025 · b40188f · b40188f
1 parent f63857d
commit b40188f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/shared/api/speech_features.hpp b/shared/api/speech_features.hpp
@@ -623,7 +623,13 @@ class Phi4AudioEmbed {
 
     SpeechLibLogMel logmel;
     // already checked in Init
-    logmel.Init(sr_val == 8000? logmel_8k_attrs_: logmel_attrs_);
+
+    // Currently we only support 8k and 16k Hz sampling rate.
+    if (sr_val != 8000 && sr_val != 16000){
+      ORTX_CXX_API_THROW("ort-extensions internal error: currently only 8k and 16k Hz sampling rate is supported", ORT_RUNTIME_EXCEPTION);
+    }
+
+    logmel.Init(sr_val == 8000 ? logmel_8k_attrs_: logmel_attrs_);
     status = logmel.Compute(stft_norm, ts_logmel);
     if (!status.IsOk()) {
       return status;