Add C++ and Python API for FireRedASR AED models (#1867)

2025-02-16 22:45:24 +08:00
parent 2337169ee2
commit 316424b382
20 changed files with 1019 additions and 26 deletions
--- a/sherpa-onnx/csrc/offline-fire-red-asr-model-meta-data.h
+++ b/sherpa-onnx/csrc/offline-fire-red-asr-model-meta-data.h
@@ -0,0 +1,28 @@
+// sherpa-onnx/csrc/offline-fire-red-asr-model-meta-data.h
+//
+// Copyright (c)  2025  Xiaomi Corporation
+#ifndef SHERPA_ONNX_CSRC_OFFLINE_FIRE_RED_ASR_MODEL_META_DATA_H_
+#define SHERPA_ONNX_CSRC_OFFLINE_FIRE_RED_ASR_MODEL_META_DATA_H_
+
+#include <string>
+#include <unordered_map>
+#include <vector>
+
+namespace sherpa_onnx {
+
+struct OfflineFireRedAsrModelMetaData {
+  int32_t sos_id;
+  int32_t eos_id;
+  int32_t max_len;
+
+  int32_t num_decoder_layers;
+  int32_t num_head;
+  int32_t head_dim;
+
+  std::vector<float> mean;
+  std::vector<float> inv_stddev;
+};
+
+}  // namespace sherpa_onnx
+
+#endif  // SHERPA_ONNX_CSRC_OFFLINE_FIRE_RED_ASR_MODEL_META_DATA_H_